Update pinned dependencies (#853)

* Remove all pins on scikit-learn
MontrealCorpusTools · Dec 2, 2024 · 0fc9b47 · 0fc9b47
1 parent 04f0b26
commit 0fc9b47
Show file tree

Hide file tree

Showing 7 changed files with 7 additions and 7 deletions.
diff --git a/ci/docker_environment.yaml b/ci/docker_environment.yaml
@@ -14,7 +14,7 @@ dependencies:
   - kaldi=*=*cpu*
   - pynini
   - openfst=1.8.3
-  - scikit-learn<1.3
+  - scikit-learn
   - hdbscan
   - baumwelch
   - ngram

diff --git a/environment.yml b/environment.yml
@@ -16,7 +16,7 @@ dependencies:
   - scipy
   - pynini
   - openfst=1.8.3
-  - scikit-learn<1.3
+  - scikit-learn
   - hdbscan
   - baumwelch
   - ngram

diff --git a/github_environment.yml b/github_environment.yml
@@ -13,7 +13,7 @@ dependencies:
   - scipy
   - pynini
   - openfst=1.8.3
-  - scikit-learn<1.3
+  - scikit-learn
   - hdbscan
   - baumwelch
   - ngram

diff --git a/montreal_forced_aligner/alignment/multiprocessing.py b/montreal_forced_aligner/alignment/multiprocessing.py
@@ -1144,7 +1144,7 @@ def _run(self):
                                 actual_score = phone_likes[i, phones[pi.phone.phone]]
                                 scores.append(phone_likes[i, top_phone_ind] - actual_score)
                         average_score = statistics.mean(scores)
-                        interval_mappings.append({"id": pi.id, "phone_goodness": average_score})
+                        interval_mappings.append({"id": pi.id, "phone_goodness": float(average_score)})
                     self.callback(interval_mappings)
                     interval_mappings = []
 

diff --git a/requirements.txt b/requirements.txt
@@ -4,7 +4,7 @@ pyyaml
 librosa
 numpy
 scipy
-scikit-learn<1.3
+scikit-learn
 requests
 biopython==1.79
 dataclassy

diff --git a/rtd_environment.yml b/rtd_environment.yml
@@ -14,7 +14,7 @@ dependencies:
   - pgvector
   - pgvector-python
   - postgresql
-  - scikit-learn<1.3
+  - scikit-learn
   - hdbscan
   - psycopg2
   - biopython=1.79

diff --git a/setup.cfg b/setup.cfg
@@ -45,7 +45,7 @@ install_requires =
     requests
     rich
     rich-click
-    scikit-learn<1.3
+    scikit-learn
     seaborn
     sqlalchemy>=1.4
     tqdm