From bb2926591dcab344c330c931286ab65d31336736 Mon Sep 17 00:00:00 2001 From: Tommy Li Date: Fri, 17 May 2019 13:24:12 -0700 Subject: [PATCH] fix sklearn tests with the correct model_dir (#98) --- python/sklearnserver/sklearnserver/test_model.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/python/sklearnserver/sklearnserver/test_model.py b/python/sklearnserver/sklearnserver/test_model.py index 64c974132e8..68605518900 100644 --- 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] \ No newline at end of file + assert response == [0]