Fixed post run and end execution

sola-st · Sep 9, 2024 · 21301bf · 21301bf
1 parent a132327
commit 21301bf
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 3 deletions.
diff --git a/src/dynapyt/post_run.py b/src/dynapyt/post_run.py
@@ -4,8 +4,9 @@
 from .utils.runtimeUtils import gather_coverage, gather_output
 
 
-def post_run(coverage_dir: str, output_dir: str):
-    gather_output(Path(output_dir))
+def post_run(coverage_dir: str = "", output_dir: str = ""):
+    if len(output_dir) > 0:
+        gather_output(Path(output_dir))
     if len(coverage_dir) > 0:
         gather_coverage(Path(coverage_dir))
 

diff --git a/src/dynapyt/runtime.py b/src/dynapyt/runtime.py
@@ -52,7 +52,7 @@ def __init__(self):
     def __del__(self):
         self.end_execution()
 
-    def end_execution(self):
+    def end_execution(self, *args):
         if self.end_execution_called:
             return
         self.end_execution_called = True