979221f
979221fc059916189fea35c787161b057ca4405c2c9df2f
658f625| Status | File | + | - |
|---|---|---|---|
| M |
tests/unit/test_mlx_convert.py
|
11 | 2 |
tests/unit/test_mlx_convert.pymodified@@ -18,9 +18,18 @@ from pathlib import Path | ||
| 18 | 18 | |
| 19 | 19 | import numpy as np |
| 20 | 20 | import pytest |
| 21 | -from safetensors.numpy import load_file, save_file | |
| 22 | 21 | |
| 23 | -from dlm_sway.backends._mlx_convert import ( | |
| 22 | +# safetensors ships in the [hf] extra (not [dev]). The fast lane runs | |
| 23 | +# without [hf]; skip the whole module when missing rather than fail | |
| 24 | +# collection. Local + slow-lane runs install [hf] and exercise these. | |
| 25 | +safetensors_numpy = pytest.importorskip( | |
| 26 | + "safetensors.numpy", | |
| 27 | + reason="safetensors not installed (install via the [hf] extra)", | |
| 28 | +) | |
| 29 | +load_file = safetensors_numpy.load_file | |
| 30 | +save_file = safetensors_numpy.save_file | |
| 31 | + | |
| 32 | +from dlm_sway.backends._mlx_convert import ( # noqa: E402 — import-after-skip | |
| 24 | 33 | MlxConvertError, |
| 25 | 34 | _extract_layer_index, |
| 26 | 35 | _strip_layer_prefix, |