From 6a8ddf15e202f96e274053614d0bbdbdee16d6a7 Mon Sep 17 00:00:00 2001 From: Emma Date: Mon, 25 Sep 2023 10:10:32 +0800 Subject: [PATCH] Remove the inference related test (#1072) --- .../examples/test_getting_started_hugectr.py | 18 ------------------ 1 file changed, 18 deletions(-) diff --git a/tests/unit/examples/test_getting_started_hugectr.py b/tests/unit/examples/test_getting_started_hugectr.py index be4820222..d0a41d816 100644 --- a/tests/unit/examples/test_getting_started_hugectr.py +++ b/tests/unit/examples/test_getting_started_hugectr.py @@ -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))))