modelscope/FunASR
1
0
Fork 0
mirror of https://github.com/modelscope/FunASR.git synced 2026-10-12 02:19:18 +00:00
Code Issues Packages Projects Releases Wiki Activity

test: avoid Python 3.11-only unittest cleanup API

Browse Source
This commit is contained in:
damnwenxi
2026-09-14 18:42:44 +08:00
parent aa353d6d22
commit eee63614f0
1 changed files with 9 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files
tests/test_submodel_device.py
+9 -6
View File
@@ -51,18 +51,21 @@ class RecordingModel(torch.nn.Module):
class TestSubmodelDevice(unittest.TestCase):
def setUp(self):
self.old_threads = torch.get_num_threads()
self.addCleanup(torch.set_num_threads, self.old_threads)
registry = {f"device-test-{role}": RecordingModel for role in ("asr", "vad", "punc", "spk")}
self.enterContext(patch.dict(tables.model_classes, registry))
self.enterContext(patch("torch.cuda.is_available", return_value=True))
self.enterContext(patch("funasr.auto.auto_model.ClusterBackend"))
self.start_patch(patch.dict(tables.model_classes, registry))
self.start_patch(patch("torch.cuda.is_available", return_value=True))
self.start_patch(patch("funasr.auto.auto_model.ClusterBackend"))
# Reject accidental model-hub access. model_conf={} uses the local registry.
self.enterContext(
self.start_patch(
patch("funasr.auto.auto_model.download_model", side_effect=AssertionError("network"))
)
self.audio = np.zeros(16000, dtype=np.float32)
def tearDown(self):
torch.set_num_threads(self.old_threads)
def start_patch(self, patcher):
value = patcher.start()
self.addCleanup(patcher.stop)
return value
def make_model(self, with_vad=True, device="cuda:0", configs=None):
configs = (