fix: check rerun error

Author: adi611

pydra/engine/run_pickled.py

@@ -2,23 +2,28 @@
 import pydra
 import sys
 
-
-def run_pickled(*file_paths):
+def run_pickled(*file_paths, rerun=False):
     loaded_objects = []
 
     for file_path in file_paths:
         with open(file_path, "rb") as file:
             loaded_objects.append(pickle.load(file))
 
     if len(loaded_objects) == 1:
-        result = loaded_objects[0](rerun=False)
+        result = loaded_objects[0](rerun=rerun)
     elif len(loaded_objects) == 3:
-        result = loaded_objects[0](loaded_objects[1], loaded_objects[2], rerun=False)
+        result = loaded_objects[0](loaded_objects[1], loaded_objects[2], rerun=rerun)
     else:
         raise ValueError("Unsupported number of loaded objects")
 
     print(f"Result: {result}")
 
-
 if __name__ == "__main__":
-    run_pickled(*sys.argv[1:])
+    rerun = False  # Default value for rerun
+    file_paths = sys.argv[1:]
+
+    if "--rerun" in file_paths:
+        rerun = True
+        file_paths.remove("--rerun")
+
+    run_pickled(*file_paths, rerun=rerun)

pydra/engine/workers.py

@@ -958,6 +958,9 @@ async def exec_psij(self, runnable, rerun=False):
                 ],
             )
 
+        if rerun:
+            spec.arguments.append("--rerun")
+
         spec.stdout_path = os.path.join(cache_dir, "demo.stdout")
         spec.stderr_path = os.path.join(cache_dir, "demo.stderr")