diff --git a/conda/environments/bench_ann_cuda-118_arch-aarch64.yaml b/conda/environments/bench_ann_cuda-118_arch-aarch64.yaml index 379be62760..02fc3be17b 100644 --- a/conda/environments/bench_ann_cuda-118_arch-aarch64.yaml +++ b/conda/environments/bench_ann_cuda-118_arch-aarch64.yaml @@ -48,7 +48,7 @@ dependencies: - pyyaml - rapids-build-backend>=0.3.0,<0.4.0.dev0 - requests -- scikit-learn +- scikit-learn>=1.5 - setuptools - sysroot_linux-aarch64==2.28 - wheel diff --git a/conda/environments/bench_ann_cuda-118_arch-x86_64.yaml b/conda/environments/bench_ann_cuda-118_arch-x86_64.yaml index 2fd2287ab4..0f8dc83f34 100644 --- a/conda/environments/bench_ann_cuda-118_arch-x86_64.yaml +++ b/conda/environments/bench_ann_cuda-118_arch-x86_64.yaml @@ -51,7 +51,7 @@ dependencies: - pyyaml - rapids-build-backend>=0.3.0,<0.4.0.dev0 - requests -- scikit-learn +- scikit-learn>=1.5 - setuptools - sysroot_linux-64==2.28 - wheel diff --git a/conda/environments/bench_ann_cuda-128_arch-aarch64.yaml b/conda/environments/bench_ann_cuda-128_arch-aarch64.yaml index c785f0bc06..e60eed8667 100644 --- a/conda/environments/bench_ann_cuda-128_arch-aarch64.yaml +++ b/conda/environments/bench_ann_cuda-128_arch-aarch64.yaml @@ -44,7 +44,7 @@ dependencies: - pyyaml - rapids-build-backend>=0.3.0,<0.4.0.dev0 - requests -- scikit-learn +- scikit-learn>=1.5 - setuptools - sysroot_linux-aarch64==2.28 - wheel diff --git a/conda/environments/bench_ann_cuda-128_arch-x86_64.yaml b/conda/environments/bench_ann_cuda-128_arch-x86_64.yaml index 616ec0a33c..84fedfa9e9 100644 --- a/conda/environments/bench_ann_cuda-128_arch-x86_64.yaml +++ b/conda/environments/bench_ann_cuda-128_arch-x86_64.yaml @@ -47,7 +47,7 @@ dependencies: - pyyaml - rapids-build-backend>=0.3.0,<0.4.0.dev0 - requests -- scikit-learn +- scikit-learn>=1.5 - setuptools - sysroot_linux-64==2.28 - wheel diff --git a/conda/recipes/cuvs-bench/recipe.yaml b/conda/recipes/cuvs-bench/recipe.yaml index 70fd1e0115..cce10d6503 100644 --- a/conda/recipes/cuvs-bench/recipe.yaml +++ b/conda/recipes/cuvs-bench/recipe.yaml @@ -70,6 +70,7 @@ requirements: - libboost - libcublas - mkl + - scikit-learn>=1.5 - if: cuda_major == "11" then: cudatoolkit diff --git a/dependencies.yaml b/dependencies.yaml index b5bfcc5ffe..0489bd5523 100644 --- a/dependencies.yaml +++ b/dependencies.yaml @@ -619,7 +619,7 @@ dependencies: - pandas - pyyaml - requests - - scikit-learn + - scikit-learn>=1.5 - output_types: [conda] packages: - matplotlib-base>=3.9 diff --git a/python/cuvs_bench/pyproject.toml b/python/cuvs_bench/pyproject.toml index 6bac4b7122..65465ff910 100644 --- a/python/cuvs_bench/pyproject.toml +++ b/python/cuvs_bench/pyproject.toml @@ -24,7 +24,7 @@ dependencies = [ "pandas", "pyyaml", "requests", - "scikit-learn", + "scikit-learn>=1.5", ] # This list was generated by `rapids-dependency-file-generator`. To make changes, edit ../../dependencies.yaml and run `rapids-dependency-file-generator`. classifiers = [ "Intended Audience :: Developers",