Make use of externally defined joblib

nielstron · May 14, 2019 · 942d3eb · 942d3eb
1 parent 873fbcd
commit 942d3eb
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 1 deletion.
diff --git a/Pipfile b/Pipfile
@@ -11,6 +11,7 @@ sklearn = "*"
 wikipedia = "*"
 stemming = "*"
 num2words = "*"
+joblib = "*"
 
 [dev-packages]
 yapf = "*"

diff --git a/quantulum3/classifier.py b/quantulum3/classifier.py
@@ -14,7 +14,7 @@
 try:
     from sklearn.linear_model import SGDClassifier
     from sklearn.feature_extraction.text import TfidfVectorizer
-    from sklearn.externals import joblib
+    import joblib
     USE_CLF = True
 except ImportError:
     SGDClassifier, TfidfVectorizer = None, None

diff --git a/requirements_classifier.txt b/requirements_classifier.txt
@@ -1,5 +1,6 @@
 numpy
 scipy
 scikit-learn
+joblib
 wikipedia
 stemming