Update , enable requirement only on darwin, disable requirement on da…

…rwin
dottxt-ai · Jun 25, 2024 · 8406f44 · 8406f44
1 parent f7d525b
commit 8406f44
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/pyproject.toml b/pyproject.toml
@@ -56,10 +56,10 @@ test = [
     "beartype<0.16.0",
     "responses",
     "llama-cpp-python",
-    "mlx-lm",
+    "mlx-lm; platform_machine == 'arm64' and sys_platform == 'darwin'",
     "huggingface_hub",
     "openai>=1.0.0",
-    "vllm",
+    "vllm; sys_platform != 'darwin'",
     "torch",
     "transformers",
 ]