From af38cb327b9a2d5d85cc4131e043930666044442 Mon Sep 17 00:00:00 2001 From: bojunliu0818 Date: Sun, 17 Apr 2022 11:30:01 -0500 Subject: [PATCH] Update test_utils: changed joblib sklearn.externals.joblib is removed in scikit-learn v0.23 --- msmbuilder/tests/test_utils.py | 6 +++++- 1 file changed, 5 insertions(+), 1 deletion(-) diff --git a/msmbuilder/tests/test_utils.py b/msmbuilder/tests/test_utils.py index 7b9ac3ba..f7b0829b 100644 --- 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