Merge pull request #268 from nzw0301/fix-numpy-for-catboost

Install `numpy<2.0.0` for catboot example
optuna · Jun 26, 2024 · 715b6b6 · 715b6b6
2 parents 8686763 + a9f5388
commit 715b6b6
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/catboost/requirements.txt b/catboost/requirements.txt
@@ -1,3 +1,4 @@
 catboost>=0.26
-numpy
+# TODO(nzw0301): Remove this numpy version constraint when catboost support numpy 2.0, which is tracked by https://github.com/catboost/catboost/issues/2671.
+numpy<2.0.0
 scikit-learn