[fix] error when read hive config to evaluate (#158)

alibaba · Apr 16, 2022 · cb93ac4 · cb93ac4
1 parent 26cad57
commit cb93ac4
Showing 1 changed file with 1 addition and 5 deletions.
diff --git a/easy_rec/python/main.py b/easy_rec/python/main.py
@@ -361,11 +361,7 @@ def evaluate(pipeline_config,
     else:
       pipeline_config.eval_input_path = eval_data_path
   train_config = pipeline_config.train_config
-
-  if pipeline_config.WhichOneof('eval_path') == 'kafka_eval_input':
-    eval_data = pipeline_config.kafka_eval_input
-  else:
-    eval_data = pipeline_config.eval_input_path
+  eval_data = _get_input_object_by_name(pipeline_config, 'eval')
 
   server_target = None
   if 'TF_CONFIG' in os.environ: