Skip to content

Commit 6a8ddf1

Browse files
authored
Remove the inference related test (#1072)
1 parent 64966c5 commit 6a8ddf1

File tree

1 file changed

+0
-18
lines changed

1 file changed

+0
-18
lines changed

tests/unit/examples/test_getting_started_hugectr.py

Lines changed: 0 additions & 18 deletions
Original file line numberDiff line numberDiff line change
@@ -80,21 +80,3 @@ def test_func():
8080
os.system("python train_hugeCTR.py")
8181
tb3.execute_cell(list(range(21, len(tb3.cells))))
8282

83-
with testbook(
84-
REPO_ROOT
85-
/ "examples/getting-started-movielens/04-Triton-Inference-with-HugeCTR.ipynb",
86-
execute=False,
87-
) as tb4:
88-
tb4.inject(
89-
f"""
90-
import os
91-
os.environ["INPUT_DATA_DIR"] = "{INPUT_DATA_DIR}"
92-
"""
93-
)
94-
tb4.execute_cell(list(range(0, 13)))
95-
with run_triton_server(
96-
os.path.join(INPUT_DATA_DIR, "model"),
97-
grpc_port=8001,
98-
backend_config=f"hugectr,ps={MODEL_DIR}/ps.json",
99-
):
100-
tb4.execute_cell(list(range(13, len(tb4.cells))))

0 commit comments

Comments
 (0)