Unpin numpy (#452)

* Unpin numpy, pin keras to latest release with numpy 2 compatiblity * Test pinnning numpy/torch for x86 mac * Pin numpy<2.0 for windows
brainglobe · Aug 21, 2024 · b39e641 · b39e641
1 parent de4d498
commit b39e641
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/pyproject.toml b/pyproject.toml
@@ -27,10 +27,12 @@ dependencies = [
     "fancylog>=0.0.7",
     "natsort",
     "numba",
-    "numpy<2",
+    "numpy<2.0.0; sys_platform=='darwin' and platform_machine=='x86_64' or sys_platform=='win32'",
+    "numpy",
     "scikit-image",
     "scikit-learn",
-    "keras>=3.0.0",
+    "keras>=3.5.0",
+    "torch<2.3.0; sys_platform=='darwin' and platform_machine=='x86_64'",
     "torch>=2.1.0",
     "tifffile",
     "tqdm",