refactor

ihmeuw · Sep 20, 2024 · 71dfcc6 · 71dfcc6
1 parent 278ee92
commit 71dfcc6
Show file tree

Hide file tree

Showing 4 changed files with 300 additions and 222 deletions.
diff --git a/src/easylink/configuration.py b/src/easylink/configuration.py
@@ -176,6 +176,7 @@ def _get_schema(self) -> Optional[PipelineSchema]:
                 errors[PIPELINE_ERRORS_KEY][schema.name] = logs
                 pass  # try the next schema
             else:
+                schema.configure_step(self.pipeline)
                 return schema
         # No schemas were validated
         exit_with_validation_error(dict(errors))

diff --git a/src/easylink/pipeline_schema.py b/src/easylink/pipeline_schema.py
@@ -18,11 +18,15 @@ class PipelineSchema(CompositeStep):
     def __repr__(self) -> str:
         return f"PipelineSchema.{self.name}"
 
+    def set_step_config(self, parent_config: LayeredConfigTree) -> None:
+        self._config = parent_config
+
     def get_pipeline_graph(self, pipeline_config: LayeredConfigTree) -> nx.MultiDiGraph:
         """Resolve the PipelineSchema into a PipelineGraph."""
+        self._config = pipeline_config
         graph = nx.MultiDiGraph()
         graph.add_node(self.name, step=self)
-        self.update_implementation_graph(graph, pipeline_config)
+        self.update_implementation_graph(graph)
         return graph
 
     @property