pin numpy < 2 [cd build]

scikit-learn · Jan 18, 2024 · 9e38cd0 · 9e38cd0
1 parent 46b5f54
commit 9e38cd0
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/setup.py b/setup.py
@@ -614,6 +614,12 @@ def setup_package():
         },
     )
 
+    # Overwrite the dependencies to not allow for NumPy >= 2.0
+    metadata["install_requires"] = [
+        f"{dep},<2.0" if dep.startswith("numpy") else dep
+        for dep in metadata["install_requires"]
+    ]
+
     commands = [arg for arg in sys.argv[1:] if not arg.startswith("-")]
     if not all(
         command in ("egg_info", "dist_info", "clean", "check") for command in commands