add flags conditions for prediction

kamalkraj · Nov 6, 2019 · 91db78c · 91db78c
1 parent 83bb8f8
commit 91db78c
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/run_classifer.py b/run_classifer.py
@@ -353,6 +353,8 @@ def metric_fn():
   if FLAGS.do_predict:
 
     logging.info("***** Running prediction*****")
+    flags.mark_flag_as_required("input_data_dir")
+    flags.mark_flag_as_required("predict_data_path")
     tokenizer = tokenization.FullTokenizer(
         vocab_file=None,spm_model_file=FLAGS.spm_model_file, do_lower_case=FLAGS.do_lower_case)