Merge pull request #8 from Hironsan/fix/#7

Update model and preprocess files
Hironsan · Jul 29, 2020 · 1277b8d · 1277b8d
2 parents 305a7b3 + 2e71b2e
commit 1277b8d
Show file tree

Hide file tree

Showing 4 changed files with 3 additions and 3 deletions.
diff --git a/MANIFEST.in b/MANIFEST.in
@@ -1 +1 @@
-recursive-include asari/data *.pkl
+recursive-include asari/data *.joblib
diff --git a/asari/api.py b/asari/api.py
@@ -16,8 +16,8 @@ class Sonar(object):
 
     def __init__(self):
         BASE_DIR = os.path.join(os.path.dirname(__file__), './data')
-        model_file = os.path.join(BASE_DIR, 'model.pkl')
-        preprocessor_file = os.path.join(BASE_DIR, 'preprocess.pkl')
+        model_file = os.path.join(BASE_DIR, 'model.joblib')
+        preprocessor_file = os.path.join(BASE_DIR, 'preprocess.joblib')
         self.estimator = joblib.load(model_file)
         self.preprocessor = joblib.load(preprocessor_file)
 

diff --git a/asari/data/model.joblib b/asari/data/model.joblib
diff --git a/asari/data/preprocess.joblib b/asari/data/preprocess.joblib
Original file line number	Diff line number	Diff line change
		@@ -1 +1 @@
		recursive-include asari/data *.pkl
		recursive-include asari/data *.joblib