Revert "Temporarily cap scikit-learn<=1.5.2 (#5)"

This reverts commit f81682d.
dmlc · Dec 10, 2024 · 8508a80 · 8508a80
1 parent f81682d
commit 8508a80
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 4 deletions.
diff --git a/containers/conda_env/aarch64_test.yml b/containers/conda_env/aarch64_test.yml
@@ -9,7 +9,7 @@ dependencies:
 - pytest-cov
 - numpy
 - scipy
-- scikit-learn<=1.5.2
+- scikit-learn
 - pandas
 - matplotlib
 - dask

diff --git a/containers/conda_env/linux_cpu_test.yml b/containers/conda_env/linux_cpu_test.yml
@@ -14,7 +14,7 @@ dependencies:
 - pylint
 - numpy
 - scipy
-- scikit-learn>=1.4.1,<=1.5.2
+- scikit-learn>=1.4.1
 - pandas
 - matplotlib
 - dask<=2024.10.0

diff --git a/containers/dockerfile/Dockerfile.gpu b/containers/dockerfile/Dockerfile.gpu
@@ -34,9 +34,8 @@ RUN \
         "nccl>=${NCCL_SHORT_VER}" \
         "dask<=2024.10.0" \
         "distributed<=2024.10.0" \
-        "scikit-learn<=1.5.2" \
         "dask-cuda=$RAPIDS_VERSION.*" "dask-cudf=$RAPIDS_VERSION.*" cupy \
-        numpy pytest pytest-timeout scipy pandas matplotlib wheel \
+        numpy pytest pytest-timeout scipy scikit-learn pandas matplotlib wheel \
         python-kubernetes urllib3 graphviz hypothesis loky \
         "pyspark>=3.4.0" cloudpickle cuda-python && \
     mamba clean --all --yes