Make mlx mandatory dependency

amazon-science · Apr 5, 2024 · d9034a1 · d9034a1
1 parent 5740e8c
commit d9034a1
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/pyproject.toml b/pyproject.toml
@@ -7,12 +7,12 @@ dependencies = [
     "torch~=2.1",         # package was tested on 2.2
     "transformers~=4.31",
     "accelerate",
+    "mlx~=0.9.0"
 ]
 
 [project.optional-dependencies]
 test = ["pytest~=8.0", "numpy~=1.21"]
 typecheck = ["mypy~=1.9"]
-mlx = ["mlx~=0.9.0"]
 
 [tool.mypy]
 ignore_missing_imports = true