SUMIN/HuggingFace_transformer
1
0
Fork 0
Code Issues Pull Requests Actions 7 Packages Projects Releases Wiki Activity

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

Browse Source 
* add more tests

* remove 2 tests

* add more tests
This commit is contained in:
Fanli Lin
2024-03-19 20:44:30 +08:00
committed by GitHub
parent f6261d7d81
commit 229ac72b1e
1 changed files with 10 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
conftest.py
View File

@@ -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 = {
"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