@@ -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