FIX: Not all Tasks have input_spec class attributes

nipype · Mar 2, 2021 · 2c0046f · 2c0046f
1 parent f0bf241
commit 2c0046f
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/pydra/engine/core.py b/pydra/engine/core.py
@@ -215,7 +215,7 @@ 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", self.input_spec)
+        input_spec = state.get("input_spec", getattr(self, "input_spec", None))
         state["inputs"] = make_klass(input_spec)(**state["inputs"])
         self.__dict__.update(state)