Attempt to fix bug with inputs/outputs shortcut

Barski-lab · Sep 21, 2020 · 52f8261 · 52f8261
1 parent ce87759
commit 52f8261
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/cwl_airflow/extensions/cwldag.py b/cwl_airflow/extensions/cwldag.py
@@ -136,7 +136,10 @@ def __assemble(self):
 
         for _, output_data in get_items(self.workflow_tool["outputs"]):
             for output_source_id, _ in get_items(output_data["outputSource"]):            # in case "outputSource" is a list
-                self.gatherer.set_upstream(task_by_out_id[output_source_id])              # connected to gatherer
+                try:
+                    self.gatherer.set_upstream(task_by_out_id[output_source_id])          # connected to another step
+                except KeyError:
+                    self.gatherer.set_upstream(self.dispatcher)                           # connected to dispatcher
 
         # safety measure in case of very specific workflows
         # if gatherer happened to be not connected to anything, connect it to all "leaves"