Cap scikit-learn<=1.5.2

hcho3 · Dec 10, 2024 · 47a764e · 47a764e
1 parent 63d097b
commit 47a764e
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 3 deletions.
diff --git a/ops/conda_env/macos_cpu_test.yml b/ops/conda_env/macos_cpu_test.yml
@@ -11,7 +11,7 @@ dependencies:
 - numpy
 - scipy
 - llvm-openmp
-- scikit-learn>=1.4.1
+- scikit-learn>=1.4.1,<=1.5.2
 - pandas
 - matplotlib
 - dask<=2024.10.0

diff --git a/ops/conda_env/win64_test.yml b/ops/conda_env/win64_test.yml
@@ -6,7 +6,7 @@ dependencies:
 - numpy
 - scipy
 - matplotlib
-- scikit-learn
+- scikit-learn<=1.5.2
 - pandas
 - pytest
 - boto3

diff --git a/ops/pipeline/test-python-wheel.sh b/ops/pipeline/test-python-wheel.sh
@@ -20,7 +20,7 @@ else
 fi
 
 source ops/pipeline/get-docker-registry-details.sh
-CONTAINER_TAG="${DOCKER_REGISTRY_URL}/${container_id}:main"
+CONTAINER_TAG="${DOCKER_REGISTRY_URL}/${container_id}:PR-5"
 
 set -x
 python3 ops/docker_run.py --container-tag "${CONTAINER_TAG}" ${gpu_option} \
-Original file line number
+Diff line change
@@ Expand Up / @@ -6,7 +6,7 @@ dependencies: @@
     - numpy
     - scipy
     - matplotlib
-    - scikit-learn
+    - scikit-learn<=1.5.2
     - pandas
     - pytest
     - boto3
@@ Expand Down @@