enable pickling of cwltool.workflow.Workflow

Author: mr-c

cwltool/workflow.py

@@ -17,6 +17,7 @@
 )
 from uuid import UUID
 
+from mypy_extensions import mypyc_attr
 from ruamel.yaml.comments import CommentedMap
 from schema_salad.exceptions import ValidationException
 from schema_salad.sourceline import SourceLine, indent
@@ -66,6 +67,7 @@ def default_make_tool(toolpath_object: CommentedMap, loadingContext: LoadingCont
 context.default_make_tool = default_make_tool
 
 
+@mypyc_attr(serializable=True)
 class Workflow(Process):
     def __init__(
         self,

tests/test_subclass_mypyc.py

@@ -12,6 +12,7 @@
 
 from cwltool.builder import Builder
 from cwltool.command_line_tool import CommandLineTool, ExpressionTool
+from cwltool.workflow import Workflow
 from cwltool.context import LoadingContext, RuntimeContext
 from cwltool.stdfsaccess import StdFsAccess
 from cwltool.update import INTERNAL_VERSION
@@ -40,6 +41,14 @@ class TestExprTool(ExpressionTool):
     a = TestExprTool(snippet, LoadingContext())
     assert a.test is False
 
+@pytest.mark.parametrize("snippet", snippet)
+def test_pickle_unpickle_workflow(snippet: CommentedMap) -> None:
+    """We can pickle & unpickle a Workflow."""
+
+    a = Workflow(snippet, LoadingContext())
+    stream = pickle.dumps(a)
+    assert stream
+    assert pickle.loads(stream)
 
 def test_serialize_builder() -> None:
     """We can pickle Builder."""