Version bumps for dependencies that don't break things because of sci…

…kit-learn deprecations
UMassCDS · Mar 11, 2024 · 001924b · 001924b
1 parent 07a5826
commit 001924b
Show file tree

Hide file tree

Showing 3 changed files with 5 additions and 5 deletions.
diff --git a/.github/workflows/python_package.yml b/.github/workflows/python_package.yml
@@ -12,7 +12,7 @@ jobs:
     runs-on: ubuntu-latest
     strategy:
       matrix:
-        python-version: ["3.8", "3.9"]
+        python-version: ["3.8", "3.9", "3.10"]
 
     steps:
       - uses: actions/checkout@v3

diff --git a/setup.cfg b/setup.cfg
@@ -16,12 +16,12 @@ install_requires =
     joblib>=1.1.0
     numpy>=1.21.0
     matplotlib==3.5.0
-    pandas==1.3.5
+    pandas>=1.3.5
     pyspark>=3.2.0
     pytimeparse==1.1.8
     scipy
     s3fs[boto3]>=2022.3.0
-    scikit-learn>=1.4.0
+    scikit-learn>=1.2.0,<1.4.0
 
 [options.extras_require]
 app =

diff --git a/tests/test_clustering.py b/tests/test_clustering.py
@@ -41,7 +41,7 @@ def text_features(spark):
 def test_clustering_model(vector_data):
     model = ic.ClusteringModel(
         vector_data,
-        KMeans(n_clusters=2, max_iter=10),
+        KMeans(n_clusters=2, max_iter=10, n_init=10),
         "test",
         {0: "AskReddit", 1: "aww", 2: "NBA"},
     )
@@ -53,7 +53,7 @@ def test_clustering_model(vector_data):
     )
 
     expected_params = {
-        "algorithm": "auto",
+        "algorithm": "lloyd",
         "copy_x": True,
         "init": "k-means++",
         "max_iter": 10,