checkpoint

AdaptiveMotorControlLab · Dec 17, 2024 · 3e4ee86 · 3e4ee86
1 parent 2ca1163
commit 3e4ee86
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/cebra/integrations/sklearn/utils.py b/cebra/integrations/sklearn/utils.py
@@ -29,6 +29,7 @@
 
 from packaging import version
 from sklearn import __version__ as sklearn_version
+sklearn_version = version.parse(sklearn_version)
 
 
 def update_old_param(old: dict, new: dict, kwargs: dict, default) -> tuple: