Skip to content

Commit

Permalink
[workflow] Skip v2-model-tester (#18)
Browse files Browse the repository at this point in the history
  • Loading branch information
Eyal-Danieli authored Jun 30, 2024
1 parent a6dfb24 commit 4c5f7de
Showing 1 changed file with 7 additions and 6 deletions.
13 changes: 7 additions & 6 deletions newflow.py
Original file line number Diff line number Diff line change
Expand Up @@ -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"]},
# )

0 comments on commit 4c5f7de

Please sign in to comment.