From 229ac72b1e941551b6650c3abc06a2a4c8bf4763 Mon Sep 17 00:00:00 2001
From: Fanli Lin <fanli.lin@intel.com>
Date: Tue, 19 Mar 2024 20:44:30 +0800
Subject: [PATCH] [tests] add more tests to `NOT_DEVICE_TESTS`  (#29670)

* add more tests

* remove 2 tests

* add more tests
---
 conftest.py | 10 ++++++++++
 1 file changed, 10 insertions(+)

diff --git a/conftest.py b/conftest.py
index 74220895aa..11e2b29756 100644
--- a/conftest.py
+++ b/conftest.py
@@ -30,9 +30,16 @@ NOT_DEVICE_TESTS = {
     "test_tokenization",
     "test_processor",
     "test_processing",
+    "test_beam_constraints",
+    "test_configuration_utils",
+    "test_data_collator",
+    "test_trainer_callback",
+    "test_trainer_utils",
     "test_feature_extraction",
     "test_image_processing",
     "test_image_processor",
+    "test_image_transforms",
+    "test_optimization",
     "test_retrieval",
     "test_config",
     "test_from_pretrained_no_checkpoint",
@@ -62,6 +69,9 @@ NOT_DEVICE_TESTS = {
     "test_flax_from_pt_safetensors",
     "ModelTest::test_pipeline_",  # None of the pipeline tests from PipelineTesterMixin (of which XxxModelTest inherits from) are running on device
     "ModelTester::test_pipeline_",
+    "/repo_utils/",
+    "/utils/",
+    "/tools/",
 }
 
 # allow having multiple repository checkouts and not needing to remember to rerun