diff --git a/tests/speech_test.py b/tests/speech_test.py index 4b8024a..cea8230 100644 --- 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")