Skip to content

Commit 465fb4e

Browse files
authored
Inference Integration test fix (#1627)
* skip download step, that is not allowed in CI * fix inference data creation path helper
1 parent 6c32ec3 commit 465fb4e

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

tests/integration/test_movielens.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -128,7 +128,7 @@ def test_movielens_tf(asv_db, bench_info, tmpdir, devices):
128128
"""
129129
)
130130
tb_train_tf.execute_cell(list(range(0, len(tb_train_tf.cells))))
131-
create_movielens_inference_data(INFERENCE_MULTI_HOT, DATA_DIR, input_path, 100)
131+
create_movielens_inference_data(INFERENCE_MULTI_HOT, input_path, 100)
132132
with test_utils.run_triton_server(
133133
INFERENCE_MULTI_HOT,
134134
"movielens",
@@ -171,7 +171,7 @@ def test_movielens_torch(asv_db, bench_info, tmpdir, devices):
171171
tb_train_torch.execute_cell(list(range(0, len(tb_train_torch.cells))))
172172

173173

174-
def create_movielens_inference_data(model_dir, data_dir, output_dir, nrows):
174+
def create_movielens_inference_data(model_dir, output_dir, nrows):
175175
import glob
176176

177177
import cudf
@@ -182,7 +182,7 @@ def create_movielens_inference_data(model_dir, data_dir, output_dir, nrows):
182182

183183
workflow_path = os.path.join(os.path.expanduser(model_dir), "movielens_nvt/1/workflow")
184184
model_path = os.path.join(os.path.expanduser(model_dir), "movielens_tf/1/model.savedmodel")
185-
data_path = os.path.join(os.path.expanduser(data_dir), "movielens/data/valid.parquet")
185+
data_path = os.path.join(os.path.expanduser(output_dir), "valid.parquet")
186186
output_dir = os.path.join(os.path.expanduser(output_dir), "movielens/")
187187
os.makedirs(output_dir)
188188
workflow_output_test_file_name = "test_inference_movielens_data.csv"

0 commit comments

Comments
 (0)