From 001924b417616d30fe5d1322960195139c2d4227 Mon Sep 17 00:00:00 2001
From: Virginia Partridge <vcpartridge@gmail.com>
Date: Mon, 11 Mar 2024 17:25:14 -0400
Subject: [PATCH] Version bumps for dependencies that don't break things
 because of scikit-learn deprecations

---
 .github/workflows/python_package.yml | 2 +-
 setup.cfg                            | 4 ++--
 tests/test_clustering.py             | 4 ++--
 3 files changed, 5 insertions(+), 5 deletions(-)

diff --git a/.github/workflows/python_package.yml b/.github/workflows/python_package.yml
index dae21f55..117b9e6c 100644
--- 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
index 9138f61b..f4fcbe65 100644
--- 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
index ba2e378d..248a71dd 100644
--- 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,