Fix path for test

caraml-dev · Jan 15, 2024 · 419ef18 · 419ef18
1 parent 412bf20
commit 419ef18
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/python/pyfunc-server/test/test_examples.py b/python/pyfunc-server/test/test_examples.py
@@ -60,16 +60,16 @@ def _get_local_endpoint(model_full_name, port):
 
 @pytest.mark.local_server_test
 def test_examples_iris():
-    XGB_PATH = os.path.join("examples/iris/models/", "model_1.bst")
-    SKLEARN_PATH = os.path.join("examples/iris/models/", "model_2.joblib")
+    XGB_PATH = os.path.join("examples/iris_http/models/", "model_1.bst")
+    SKLEARN_PATH = os.path.join("examples/iris_http/models/", "model_2.joblib")
 
     port = _get_free_port()
 
     p = Process(
         target=merlin.run_pyfunc_model,
         kwargs={
             "model_instance": IrisModel(),
-            "conda_env": "examples/iris/env.yaml",
+            "conda_env": "examples/iris_http/env.yaml",
             "code_dir": ["examples"],
             "artifacts": {
                 "xgb_model": XGB_PATH,