fix sklearn tests with the correct model_dir (kubeflow#98)

magdalenakuhn17 · May 17, 2019 · bb29265 · bb29265
1 parent 667bcef
commit bb29265
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/python/sklearnserver/sklearnserver/test_model.py b/python/sklearnserver/sklearnserver/test_model.py
@@ -4,7 +4,7 @@
 import joblib
 import os
 
-model_dir = "/path/to/kfserving/docs/samples/sklearn"
+model_dir = "../../docs/samples/sklearn"
 JOBLIB_FILE = "model.joblib"
 
 def test_model():
@@ -18,4 +18,4 @@ def test_model():
      server.load()
      request = X[0:1].tolist()
      response = server.predict(request)
-     assert response == [0]
+     assert response == [0]