mirror of
https://github.com/jamiepine/voicebox.git
synced 2026-09-16 13:20:39 -07:00
f750596364
* fix(setup): install mlx-lm and mlx-audio in setup-python on Apple Silicon The dev setup installed requirements-mlx.txt but not mlx-audio/mlx-lm themselves, so POST /transcribe failed on a fresh Apple Silicon setup with "No module named 'mlx_audio'" (then "No module named 'mlx_lm'"). The release workflow already installs both with --no-deps (they declare transformers>=5.x, conflicting with our <=4.57.x cap); mirror that in the setup-python recipe with the same pins. Co-Authored-By: Claude Fable 5 <[email protected]> * test: add MLX smoke test for the --no-deps mlx-audio/mlx-lm install mlx-audio and mlx-lm are installed --no-deps, so a missing transitive dependency only surfaces at import time. Add a pytest-discoverable smoke test (skipped off Apple Silicon) covering the exact entry points the backend uses: mlx_audio.tts.load, mlx_audio.stt.load (which also exercises the miniaudio dep from issue #505), mlx_lm.load/generate, and a basic mlx.core op. Co-Authored-By: Claude Fable 5 <[email protected]> --------- Co-authored-by: Claude Fable 5 <[email protected]>
56 lines
1.6 KiB
Python
56 lines
1.6 KiB
Python
"""
|
|
Smoke test for the MLX backend dependencies on Apple Silicon.
|
|
|
|
Guards the `--no-deps` install of mlx-audio/mlx-lm done by `just setup-python`
|
|
and release.yml: those packages skip their declared dependencies (transformers
|
|
>=5.x conflict), so a missing transitive dep only surfaces at import time.
|
|
This test fails fast if the MLX STT/TTS entry points the backend uses stop
|
|
importing (e.g. the `miniaudio` regression from issue #505).
|
|
|
|
Usage:
|
|
python -m pytest backend/tests/test_mlx_smoke.py -v
|
|
"""
|
|
|
|
import platform
|
|
import sys
|
|
|
|
import pytest
|
|
|
|
pytestmark = pytest.mark.skipif(
|
|
not (sys.platform == "darwin" and platform.machine() == "arm64"),
|
|
reason="MLX packages are only installed on Apple Silicon macOS",
|
|
)
|
|
|
|
|
|
def test_mlx_core_runs():
|
|
"""The MLX runtime itself works (Metal array op)."""
|
|
import mlx.core as mx
|
|
|
|
assert mx.array([1, 2]).sum().item() == 3
|
|
|
|
|
|
def test_mlx_audio_tts_entry_point():
|
|
"""`from mlx_audio.tts import load` — used by MLXBackend.load_model_async."""
|
|
from mlx_audio.tts import load
|
|
|
|
assert callable(load)
|
|
|
|
|
|
def test_mlx_audio_stt_entry_point():
|
|
"""`from mlx_audio.stt import load` — used by the Whisper MLX STT path.
|
|
|
|
Importing mlx_audio.stt also pulls in miniaudio, so this catches the
|
|
ModuleNotFoundError from issue #505 on fresh installs.
|
|
"""
|
|
from mlx_audio.stt import load
|
|
|
|
assert callable(load)
|
|
|
|
|
|
def test_mlx_lm_entry_points():
|
|
"""`mlx_lm.load` / `mlx_lm.generate` — used by qwen_llm_backend."""
|
|
from mlx_lm import generate, load
|
|
|
|
assert callable(load)
|
|
assert callable(generate)
|