Skip to content

Commit

Permalink
Remove the inference related test (#1072)
Browse files Browse the repository at this point in the history
  • Loading branch information
EmmaQiaoCh authored Sep 25, 2023
1 parent 64966c5 commit 6a8ddf1
Showing 1 changed file with 0 additions and 18 deletions.
18 changes: 0 additions & 18 deletions tests/unit/examples/test_getting_started_hugectr.py
Original file line number Diff line number Diff line change
Expand Up @@ -80,21 +80,3 @@ def test_func():
os.system("python train_hugeCTR.py")
tb3.execute_cell(list(range(21, len(tb3.cells))))

with testbook(
REPO_ROOT
/ "examples/getting-started-movielens/04-Triton-Inference-with-HugeCTR.ipynb",
execute=False,
) as tb4:
tb4.inject(
f"""
import os
os.environ["INPUT_DATA_DIR"] = "{INPUT_DATA_DIR}"
"""
)
tb4.execute_cell(list(range(0, 13)))
with run_triton_server(
os.path.join(INPUT_DATA_DIR, "model"),
grpc_port=8001,
backend_config=f"hugectr,ps={MODEL_DIR}/ps.json",
):
tb4.execute_cell(list(range(13, len(tb4.cells))))

0 comments on commit 6a8ddf1

Please sign in to comment.