Skip to content

Commit 4fd8bb2

Browse files
committed
replace with single function for psijworker
1 parent af84f02 commit 4fd8bb2

File tree

4 files changed

+1004
-1020
lines changed

4 files changed

+1004
-1020
lines changed

pydra/engine/run_pickled.py

Lines changed: 24 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,24 @@
1+
import pickle
2+
import pydra
3+
import sys
4+
5+
6+
def run_pickled(*file_paths):
7+
loaded_objects = []
8+
9+
for file_path in file_paths:
10+
with open(file_path, "rb") as file:
11+
loaded_objects.append(pickle.load(file))
12+
13+
if len(loaded_objects) == 1:
14+
result = loaded_objects[0](rerun=False)
15+
elif len(loaded_objects) == 3:
16+
result = loaded_objects[0](loaded_objects[1], loaded_objects[2], rerun=False)
17+
else:
18+
raise ValueError("Unsupported number of loaded objects")
19+
20+
print(f"Result: {result}")
21+
22+
23+
if __name__ == "__main__":
24+
run_pickled(*sys.argv[1:])

pydra/engine/run_pickled_function.py

Lines changed: 0 additions & 17 deletions
This file was deleted.

pydra/engine/run_pickled_function_2.py

Lines changed: 0 additions & 23 deletions
This file was deleted.

0 commit comments

Comments
 (0)