修正: 対応デバイスモックの属性不足を修正 (#1224)

fix: コアバージョンモックの属性不足を修正
VOICEVOX · May 14, 2024 · 418a9d3 · 418a9d3
1 parent 67dbf5f
commit 418a9d3
Show file tree

Hide file tree

Showing 3 changed files with 5 additions and 8 deletions.
diff --git a/test/e2e/single_api/__snapshots__/test_supported_devices/test_get_supported_devices_200.json b/test/e2e/single_api/__snapshots__/test_supported_devices/test_get_supported_devices_200.json
diff --git a/test/tts_pipeline/test_tts_engine.py b/test/tts_pipeline/test_tts_engine.py
@@ -1,3 +1,4 @@
+import json
 from test.utility import pydantic_to_native_type, round_floats
 from unittest import TestCase
 from unittest.mock import Mock
@@ -97,7 +98,7 @@ def metas(self) -> str:
         return ""
 
     def supported_devices(self) -> str:
-        return ""
+        return json.dumps({"cpu": True, "cuda": False, "dml": False})
 
     def is_model_loaded(self, style_id: str) -> bool:
         return True

diff --git a/voicevox_engine/dev/core/mock.py b/voicevox_engine/dev/core/mock.py
@@ -229,12 +229,7 @@ def sf_decode_forward(
         return np.array(result, dtype=np.float32)
 
     def supported_devices(self) -> str:
-        return json.dumps(
-            {
-                "cpu": True,
-                "cuda": False,
-            }
-        )
+        return json.dumps({"cpu": True, "cuda": False, "dml": False})
 
     def finalize(self) -> None:
         pass