Add setuptools dependency for dask-glm.

rapidsai · Sep 11, 2024 · d019dec · d019dec
1 parent 333c5ab
commit d019dec
Show file tree

Hide file tree

Showing 4 changed files with 4 additions and 0 deletions.
diff --git a/conda/environments/all_cuda-118_arch-x86_64.yaml b/conda/environments/all_cuda-118_arch-x86_64.yaml
@@ -68,6 +68,7 @@ dependencies:
 - scikit-learn==1.5
 - scipy>=1.8.0
 - seaborn
+- setuptools
 - sphinx-copybutton
 - sphinx-markdown-tables
 - sphinx<6

diff --git a/conda/environments/all_cuda-125_arch-x86_64.yaml b/conda/environments/all_cuda-125_arch-x86_64.yaml
@@ -64,6 +64,7 @@ dependencies:
 - scikit-learn==1.5
 - scipy>=1.8.0
 - seaborn
+- setuptools
 - sphinx-copybutton
 - sphinx-markdown-tables
 - sphinx<6

diff --git a/dependencies.yaml b/dependencies.yaml
@@ -515,6 +515,7 @@ dependencies:
           - statsmodels
           - umap-learn==0.5.6
           - pynndescent
+          - setuptools  # Needed on Python 3.12 for dask-glm, which requires pkg_resources but Python 3.12 doesn't have setuptools by default
       - output_types: conda
         packages:
           - pip

diff --git a/python/cuml/pyproject.toml b/python/cuml/pyproject.toml
@@ -125,6 +125,7 @@ test = [
     "pytest==7.*",
     "scikit-learn==1.5",
     "seaborn",
+    "setuptools",
     "statsmodels",
     "umap-learn==0.5.6",
 ] # This list was generated by `rapids-dependency-file-generator`. To make changes, edit ../../dependencies.yaml and run `rapids-dependency-file-generator`.