From 4c5f7debf7a266833d3091bae280af18faf66b12 Mon Sep 17 00:00:00 2001
From: Eyal Danieli <eyald@iguazio.com>
Date: Sun, 30 Jun 2024 18:34:01 +0300
Subject: [PATCH] [workflow] Skip `v2-model-tester`  (#18)

---
 newflow.py | 13 +++++++------
 1 file changed, 7 insertions(+), 6 deletions(-)

diff --git a/newflow.py b/newflow.py
index 6dbe31f..50594d6 100644
--- a/newflow.py
+++ b/newflow.py
@@ -95,10 +95,11 @@ def newpipe():
         models=[{"key": f"{DATASET}:v1", "model_path": train.outputs["model"]}],
     )
 
+    # TODO: Add the following function once v2-model-tester is fixed
     # test out new model server (via REST API calls), use imported function
-    run_function(
-        "hub://v2-model-tester",
-        name="model-tester",
-        params={"addr": deploy.outputs["endpoint"], "model": f"{DATASET}:v1"},
-        inputs={"table": train.outputs["test_set"]},
-    )
+    # run_function(
+    #     "hub://v2-model-tester",
+    #     name="model-tester",
+    #     params={"addr": deploy.outputs["endpoint"], "model": f"{DATASET}:v1"},
+    #     inputs={"table": train.outputs["test_set"]},
+    # )