FIX: Allow input_spec/output_spec to be class variables

Author: effigies

pydra/engine/core.py

@@ -198,8 +198,10 @@ def __str__(self):
 
     def __getstate__(self):
         state = self.__dict__.copy()
-        state["input_spec"] = cp.dumps(state["input_spec"])
-        state["output_spec"] = cp.dumps(state["output_spec"])
+        if "input_spec" in state:
+            state["input_spec"] = cp.dumps(state["input_spec"])
+        if "output_spec" in state:
+            state["output_spec"] = cp.dumps(state["output_spec"])
         inputs = {}
         for k, v in attr.asdict(state["inputs"]).items():
             if k.startswith("_"):
@@ -209,9 +211,12 @@ def __getstate__(self):
         return state
 
     def __setstate__(self, state):
-        state["input_spec"] = cp.loads(state["input_spec"])
-        state["output_spec"] = cp.loads(state["output_spec"])
-        state["inputs"] = make_klass(state["input_spec"])(**state["inputs"])
+        if "input_spec" in state:
+            state["input_spec"] = cp.loads(state["input_spec"])
+        if "output_spec" in state:
+            state["output_spec"] = cp.loads(state["output_spec"])
+        input_spec = state.get("input_spec", self.input_spec)
+        state["inputs"] = make_klass(input_spec)(**state["inputs"])
         self.__dict__.update(state)
 
     def __getattr__(self, name):