added definition to Result attributes that need to be cloudpickled

Author: tclose

new-docs/source/tutorial/3-troubleshooting.ipynb

@@ -128,8 +128,9 @@
     "# This workflow will fail because we are trying to divide by 0\n",
     "wf = UnsafeDivisionWorkflow(a=10, b=5).split(denominator=[3, 2 ,0])\n",
     "\n",
-    "with Submitter(worker=\"cf\") as sub:\n",
-    "    result = sub(wf)\n",
+    "if __name__ == \"__main__\":\n",
+    "    with Submitter(worker=\"cf\") as sub:\n",
+    "        result = sub(wf)\n",
     "    \n",
     "if result.errored:\n",
     "    print(\"Workflow failed with errors:\\n\" + str(result.errors))\n",

pydra/engine/specs.py

@@ -531,15 +531,19 @@ class Result(ty.Generic[OutputsType]):
     errored: bool = False
     definition: TaskDef[OutputsType] | None = None
 
+    CLOUD_PICKLE_ATTRS = ("outputs", "definition")
+
     def __getstate__(self):
         state = attrs_values(self)
-        if state["outputs"] is not None:
-            state["outputs"] = cp.dumps(state["outputs"])
+        for attr in self.CLOUD_PICKLE_ATTRS:
+            if state[attr] is not None:
+                state[attr] = cp.dumps(state[attr])
         return state
 
     def __setstate__(self, state):
-        if state["outputs"] is not None:
-            state["outputs"] = cp.loads(state["outputs"])
+        for attr in self.CLOUD_PICKLE_ATTRS:
+            if state[attr] is not None:
+                state[attr] = cp.loads(state[attr])
         for name, val in state.items():
             setattr(self, name, val)