Selaa lähdekoodia

update paddlex_cli

cuicheng01 10 kuukautta sitten
vanhempi
commit
ad43de32eb
1 muutettua tiedostoa jossa 8 lisäystä ja 6 poistoa
  1. 8 6
      paddlex/paddlex_cli.py

+ 8 - 6
paddlex/paddlex_cli.py

@@ -280,12 +280,14 @@ def main():
             interactive_get_pipeline(args.get_pipeline_config, args.save_path)
         else:
             pipeline_args_dict = {}
-            for arg in pipeline_args:
-                arg_name = arg["name"].lstrip("-")
-                if hasattr(args, arg_name):
-                    pipeline_args_dict[arg_name] = getattr(args, arg_name)
-                else:
-                    logging.warning(f"Argument {arg_name} is missing in args")
+            from .utils.flags import USE_NEW_INFERENCE
+            if USE_NEW_INFERENCE:
+                for arg in pipeline_args:
+                    arg_name = arg["name"].lstrip("-")
+                    if hasattr(args, arg_name):
+                        pipeline_args_dict[arg_name] = getattr(args, arg_name)
+                    else:
+                        logging.warning(f"Argument {arg_name} is missing in args")
             return pipeline_predict(
                 args.pipeline,
                 args.input,