Update test_utils: changed joblib

sklearn.externals.joblib is removed in scikit-learn v0.23
msmbuilder · Apr 17, 2022 · af38cb3 · af38cb3
1 parent 3840a40
commit af38cb3
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/msmbuilder/tests/test_utils.py b/msmbuilder/tests/test_utils.py
@@ -3,7 +3,11 @@
 import numpy as np
 import sklearn.pipeline
 from mdtraj.testing import eq
-from sklearn.externals.joblib import dump as jl_dump
+# sklearn.externals.joblib is removed in scikit-learn v0.23
+try:
+    from joblib import dump as jl_dump
+except ImportError:
+    from sklearn.externals.joblib import dump as jl_dump
 
 from msmbuilder.decomposition import tICA
 from msmbuilder.utils import Subsampler, dump, load