fix out_path bug (#30)

PaddlePaddle · Oct 11, 2024 · 95f18cb · 95f18cb
1 parent a7a6cfc
commit 95f18cb
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/paddleapex/apex/run_paddle.py b/paddleapex/apex/run_paddle.py
@@ -552,7 +552,7 @@ def arg_parser(parser):
     forward_content = api_json_read(cfg.json_path)
     out_path = os.path.realpath(cfg.out_path) if cfg.out_path else "./"
     if os.path.exists(out_path):
-        shutil.rmtree(out_path)
+        print_warn_log("The output path already exists and the file with the same name will be overwritten.")
     ut_case_parsing(forward_content, cfg)
     print_info_log("UT save completed")
     warning_log_pth = os.path.join(out_path, "./warning_log.txt")

diff --git a/paddleapex/apex/run_torch.py b/paddleapex/apex/run_torch.py
@@ -596,7 +596,7 @@ def arg_parser(parser):
         "-out",
         "--dump_path",
         dest="out_path",
-        default="./paddle/",
+        default="./torch/",
         type=str,
         help="<optional> The ut task result out path.",
         required=False,
@@ -645,7 +645,7 @@ def arg_parser(parser):
     forward_content = api_json_read(cfg.json_path)
     out_path = os.path.realpath(cfg.out_path) if cfg.out_path else "./"
     if os.path.exists(out_path):
-        shutil.rmtree(out_path)
+        print_warn_log("The output path already exists and the file with the same name will be overwritten.")
     ut_case_parsing(forward_content, cfg)
     print_info_log("UT save completed")
     warning_log_pth = os.path.join(out_path, "./warning_log.txt")