From 3237bb1f37ffc989cb01c31d9006043b15da7df0 Mon Sep 17 00:00:00 2001 From: Hyunsu Cho Date: Tue, 10 Dec 2024 02:16:06 -0800 Subject: [PATCH] Temporarily cap scikit-learn<=1.5.2 --- containers/conda_env/aarch64_test.yml | 2 +- containers/conda_env/linux_cpu_test.yml | 2 +- containers/dockerfile/Dockerfile.gpu | 3 ++- 3 files changed, 4 insertions(+), 3 deletions(-) diff --git a/containers/conda_env/aarch64_test.yml b/containers/conda_env/aarch64_test.yml index 14305eb..4c74160 100644 --- 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 +- scikit-learn<=1.5.2 - pandas - matplotlib - dask diff --git a/containers/conda_env/linux_cpu_test.yml b/containers/conda_env/linux_cpu_test.yml index 1ec2a54..2bc35dc 100644 --- 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 +- scikit-learn>=1.4.1,<=1.5.2 - pandas - matplotlib - dask<=2024.10.0 diff --git a/containers/dockerfile/Dockerfile.gpu b/containers/dockerfile/Dockerfile.gpu index fe831d8..f20fa01 100644 --- a/containers/dockerfile/Dockerfile.gpu +++ b/containers/dockerfile/Dockerfile.gpu @@ -34,8 +34,9 @@ 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 scikit-learn pandas matplotlib wheel \ + numpy pytest pytest-timeout scipy pandas matplotlib wheel \ python-kubernetes urllib3 graphviz hypothesis loky \ "pyspark>=3.4.0" cloudpickle cuda-python && \ mamba clean --all --yes