FIX: Only check for self.input_spec if not in state

nipype · Mar 2, 2021 · 7894d0b · 7894d0b
1 parent 60b062d
commit 7894d0b
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/pydra/engine/core.py b/pydra/engine/core.py
@@ -217,7 +217,9 @@ def __setstate__(self, 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", getattr(self, "input_spec", None))
+        input_spec = state.get("input_spec")
+        if input_spec is None:  # If it is not saved, it should be a class attribute
+            input_spec = self.input_spec
         state["inputs"] = make_klass(input_spec)(**state["inputs"])
         self.__dict__.update(state)