fix/scikit-learn dependency (#237)

* update setup.py * update requirements.txt * update doc/requirements.txt
pyRiemann · Jan 18, 2024 · 402bbd2 · 402bbd2
1 parent 9f3c71f
commit 402bbd2
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 1 deletion.
diff --git a/doc/requirements.txt b/doc/requirements.txt
@@ -5,7 +5,7 @@ numpydoc
 cython
 mne[data]>=0.24
 seaborn>=0.12.1
-scikit-learn
+scikit-learn==1.3.2
 imbalanced-learn==0.11.0
 joblib
 pandas

diff --git a/requirements.txt b/requirements.txt
@@ -1,5 +1,6 @@
 numpy<1.27
 cython
+scikit-learn==1.3.2
 git+https://github.com/pyRiemann/pyRiemann#egg=pyriemann
 qiskit_machine_learning==0.6.1
 qiskit-ibm-provider==0.7.3

diff --git a/setup.py b/setup.py
@@ -46,6 +46,7 @@
                         'scipy==1.11.4',
                         'docplex>=2.21.207',
                         'firebase_admin==6.2.0',
+                        'scikit-learn==1.3.2',
                         'tqdm'
                         ],
       extras_require={'docs': [