fix: force numpy < 2.0.0 on mac intel (#69)

Signed-off-by: Michele Dolfi <[email protected]>
DS4SD · Jan 9, 2025 · 7f9365f · 7f9365f
1 parent b1f9af1
commit 7f9365f
Show file tree

Hide file tree

Showing 2 changed files with 52 additions and 5 deletions.
diff --git a/poetry.lock b/poetry.lock
diff --git a/pyproject.toml b/pyproject.toml
@@ -25,7 +25,10 @@ python = "^3.9"
 torch = "^2.2.2"
 torchvision = "^0"
 transformers = "^4.42.0"
-numpy = ">=1.24.4,<3.0.0"
+numpy = [
+    { version = ">=1.24.4,<3.0.0", markers = 'sys_platform != "darwin" or platform_machine != "x86_64"' },
+    { version = ">=1.24.4,<2.0.0", markers = 'sys_platform == "darwin" and platform_machine == "x86_64"' },
+]
 jsonlines = "^3.1.0"
 Pillow = "^10.0.0"
 tqdm = "^4.64.0"