diff --git a/pystacknet/pystacknet.py b/pystacknet/pystacknet.py index 356f77e..4d4b967 100644 --- a/pystacknet/pystacknet.py +++ b/pystacknet/pystacknet.py @@ -19,7 +19,7 @@ from sklearn.model_selection import KFold from sklearn.utils import check_X_y,check_array,check_consistent_length, column_or_1d import inspect -from sklearn.externals.joblib import delayed,Parallel +from joblib import delayed,Parallel import operator import time from sklearn.preprocessing import LabelEncoder diff --git a/setup.py b/setup.py index 06e987e..f4dcaba 100644 --- a/setup.py +++ b/setup.py @@ -26,7 +26,8 @@ install_requires=[ 'numpy >= 1.14.0', 'scipy >= 1.1.0', - 'scikit-learn >= 0.19.1' + 'scikit-learn >= 0.19.1', + 'joblib >= 0.13.2' ], classifiers=[