diff --git a/containers/conda_env/aarch64_test.yml b/containers/conda_env/aarch64_test.yml index 4c74160..14305eb 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<=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 index 2bc35dc..1ec2a54 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,<=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 index f20fa01..fe831d8 100644 --- 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