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

Simplify running tests in a subprocess (#34213)

Browse Source 
* check

* check

* check

* check

* add docstring

---------

Co-authored-by: ydshieh <ydshieh@users.noreply.github.com>
This commit is contained in:
Yih-Dar
2024-10-29 10:48:57 +01:00
committed by GitHub
parent a1835195d1
commit 439334c8fb
3 changed files with 52 additions and 9 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
tests/models/imagegpt/test_modeling_imagegpt.py
View File

@@ -18,7 +18,7 @@ import inspect
import unittest
from transformers import ImageGPTConfig
from transformers.testing_utils import require_torch, require_vision, slow, torch_device
from transformers.testing_utils import require_torch, require_vision, run_test_using_subprocess, slow, torch_device
from transformers.utils import cached_property, is_torch_available, is_vision_available
from ...generation.test_utils import GenerationTesterMixin
@@ -257,11 +257,9 @@ class ImageGPTModelTest(ModelTesterMixin, GenerationTesterMixin, PipelineTesterM
self.assertEqual(len(scores), length)
self.assertListEqual([iter_scores.shape for iter_scores in scores], [expected_shape] * len(scores))
@unittest.skip(
reason="After #33632, this test still passes, but many subsequential tests fail with `device-side assert triggered`"
)
@run_test_using_subprocess
def test_beam_search_generate_dict_outputs_use_cache(self):
pass
super().test_beam_search_generate_dict_outputs_use_cache()
def setUp(self):
self.model_tester = ImageGPTModelTester(self)