cleanup

Author: KRRT7

codeflash/optimization/optimizer.py

@@ -407,9 +407,8 @@ def run(self) -> None:
         if self.args.worktree:
             self.worktree_mode()
 
-        test_root = Path(self.test_cfg.tests_root)
-        if test_root.exists():
-            leftover_trace_files = list(test_root.glob("*.trace"))
+        if not self.args.replay_test and self.test_cfg.tests_root.exists():
+            leftover_trace_files = list(self.test_cfg.tests_root.glob("*.trace"))
             if leftover_trace_files:
                 logger.debug(f"Cleaning up {len(leftover_trace_files)} leftover trace file(s) from previous runs")
                 cleanup_paths(leftover_trace_files)
@@ -586,9 +585,8 @@ def cleanup_temporary_paths(self) -> None:
         paths_to_cleanup = [self.test_cfg.concolic_test_root_dir, self.replay_tests_dir]
         if self.trace_file:
             paths_to_cleanup.append(self.trace_file)
-        test_root = Path(self.test_cfg.tests_root)
-        if test_root.exists():
-            for trace_file in test_root.glob("*.trace"):
+        if self.test_cfg.tests_root.exists():
+            for trace_file in self.test_cfg.tests_root.glob("*.trace"):
                 if trace_file not in paths_to_cleanup:
                     paths_to_cleanup.append(trace_file)
         cleanup_paths(paths_to_cleanup)