Skip to content

GitLab

Unverified Commit 229ac72b authored by Fanli Lin's avatar Fanli Lin Committed by GitHub
Browse files

[tests] add more tests to `NOT_DEVICE_TESTS` (#29670) 

* add more tests

* remove 2 tests

* add more tests
parent f6261d7d
Hide whitespace changes
Inline Side-by-side
Showing with 10 additions and 0 deletions
conftest.py
View file @ 229ac72b
...@@ -30,9 +30,16 @@ NOT_DEVICE_TESTS = { ...@@ -30,9 +30,16 @@ NOT_DEVICE_TESTS = {
"test_tokenization", "test_tokenization",
"test_processor", "test_processor",
"test_processing", "test_processing",
"test_beam_constraints",
"test_configuration_utils",
"test_data_collator",
"test_trainer_callback",
"test_trainer_utils",
"test_feature_extraction", "test_feature_extraction",
"test_image_processing", "test_image_processing",
"test_image_processor", "test_image_processor",
"test_image_transforms",
"test_optimization",
"test_retrieval", "test_retrieval",
"test_config", "test_config",
"test_from_pretrained_no_checkpoint", "test_from_pretrained_no_checkpoint",
...@@ -62,6 +69,9 @@ NOT_DEVICE_TESTS = { ...@@ -62,6 +69,9 @@ NOT_DEVICE_TESTS = {
"test_flax_from_pt_safetensors", "test_flax_from_pt_safetensors",
"ModelTest::test_pipeline_", # None of the pipeline tests from PipelineTesterMixin (of which XxxModelTest inherits from) are running on device "ModelTest::test_pipeline_", # None of the pipeline tests from PipelineTesterMixin (of which XxxModelTest inherits from) are running on device
"ModelTester::test_pipeline_", "ModelTester::test_pipeline_",
"/repo_utils/",
"/utils/",
"/tools/",
} }
# allow having multiple repository checkouts and not needing to remember to rerun # allow having multiple repository checkouts and not needing to remember to rerun
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment