Fix for scikit-learn version and nimbus_nlp model imports

calpoly-csai · Feb 29, 2020 · 6c7c45a · 6c7c45a
1 parent f16a7eb
commit 6c7c45a
Show file tree

Hide file tree

Showing 3 changed files with 2 additions and 2 deletions.
diff --git a/nimbus_nlp/__init__.py b/nimbus_nlp/__init__.py
diff --git a/nimbus_nlp/question_classifier.py b/nimbus_nlp/question_classifier.py
@@ -6,7 +6,7 @@
 import pandas as pd
 import sys
 import json
-from nimbus_nlp.save_and_load_model import save_model, load_latest_model, PROJECT_DIR
+from save_and_load_model import save_model, load_latest_model, PROJECT_DIR
 
 # TODO: move the functionality in this module into class(es), so that it can be more easily used as a dependency
 

diff --git a/requirements.txt b/requirements.txt
@@ -72,7 +72,7 @@ PyYAML==5.3
 regex==2020.2.20
 requests==2.23.0
 rsa==4.0
-scikit-learn==0.22.2
+scikit-learn==0.20.2
 scipy==1.4.1
 six==1.13.0
 sklearn==0.0