FIX: TransformerMixin

mne-tools · Sep 17, 2024 · 378a186 · 378a186
1 parent 647dd71
commit 378a186
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 1 deletion.
diff --git a/mne_connectivity/decoding/decomposition.py b/mne_connectivity/decoding/decomposition.py
@@ -5,10 +5,11 @@
 # License: BSD (3-clause)
 
 
+from sklearn.base import TransformerMixin
+
 import numpy as np
 from mne import Info
 from mne._fiff.pick import pick_info
-from mne.decoding.mixin import TransformerMixin
 from mne.defaults import _BORDER_DEFAULT, _EXTRAPOLATE_DEFAULT, _INTERPOLATION_DEFAULT
 from mne.evoked import EvokedArray
 from mne.fixes import BaseEstimator

diff --git a/pyproject.toml b/pyproject.toml
@@ -26,6 +26,7 @@ dependencies = [
   'scipy >= 1.4.0',
   'tqdm',
   'xarray >= 2023.11.0',
+  'scikit-learn >= 1.2'
 ]
 description = 'mne-connectivity: A module for connectivity data analysis with MNE.'
 dynamic = ["version"]