fix: piper is imported on ARM

fedirz · Nov 1, 2024 · 979b201 · 979b201
1 parent e59ebf5
commit 979b201
Showing 1 changed file with 5 additions and 3 deletions.
diff --git a/tests/speech_test.py b/tests/speech_test.py
@@ -5,7 +5,11 @@
 import pytest
 import soundfile as sf
 
-from faster_whisper_server.routers.speech import (
+platform_machine = platform.machine()
+if platform_machine != "x86_64":
+    pytest.skip("Only supported on x86_64", allow_module_level=True)
+
+from faster_whisper_server.routers.speech import (  # noqa: E402
     DEFAULT_MODEL,
     DEFAULT_RESPONSE_FORMAT,
     DEFAULT_VOICE,
@@ -15,8 +19,6 @@
 
 DEFAULT_INPUT = "Hello, world!"
 
-platform_machine = platform.machine()
-
 
 @pytest.mark.asyncio
 @pytest.mark.skipif(platform_machine != "x86_64", reason="Only supported on x86_64")