infer

recommend-model-produce/src/main/python/tools/static_ps_infer_v2.py

@@ -229,7 +229,7 @@ class Main(object):
             return
         self.config["runner.reader_type"] = self.config.get(
             "runner.reader_type", "QueueDataset")
-        self.reader, self.file_list = get_reader(self.input_data, config)
+        self.reader, self.file_list = get_infer_reader(self.input_data, config)
         self.example_nums = 0
         self.count_method = self.config.get("runner.example_count_method",
                                             "example")

recommend-model-produce/src/main/python/tools/utils/static_ps/reader_helper.py

@@ -35,7 +35,7 @@ def get_reader(input_var, config):
     assert train_data_path != ""
 
     train_data_path = os.path.join(config["config_abs_dir"], train_data_path)
-    print("test_data_path is: {}".format(test_data_path))
+
     assert reader_type in [
         "QueueDataset", "DataLoader", "RecDataset", "InmemoryDataset", None
     ]