From 4adcbaed3147d70cae8f4191ed9959941fa83759 Mon Sep 17 00:00:00 2001 From: tlento Date: Thu, 16 May 2024 15:00:55 -0700 Subject: [PATCH] Rename subgraph helper method --- metricflow/dataflow/dataflow_plan.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/metricflow/dataflow/dataflow_plan.py b/metricflow/dataflow/dataflow_plan.py index 4685ddad47..8ec255e038 100644 --- a/metricflow/dataflow/dataflow_plan.py +++ b/metricflow/dataflow/dataflow_plan.py @@ -209,13 +209,13 @@ def __init__(self, sink_nodes: Sequence[DataflowPlanNode], plan_id: Optional[Dag def sink_node(self) -> DataflowPlanNode: # noqa: D102 return self._sink_nodes[0] - def __complete_subgraph(self, node: DataflowPlanNode) -> Sequence[DataflowPlanNode]: + def __all_nodes_in_subgraph(self, node: DataflowPlanNode) -> Sequence[DataflowPlanNode]: """Node accessor for retrieving a flattened sequence of all nodes in the subgraph upstream of the input node. Useful for gathering nodes for subtype-agnostic operations, such as common property access or simple counts. """ flattened_parent_subgraphs = tuple( - more_itertools.collapse(self.__complete_subgraph(parent_node) for parent_node in node.parent_nodes) + more_itertools.collapse(self.__all_nodes_in_subgraph(parent_node) for parent_node in node.parent_nodes) ) return (node,) + flattened_parent_subgraphs @@ -225,7 +225,7 @@ def source_semantic_models(self) -> FrozenSet[SemanticModelReference]: return frozenset( [ node._input_semantic_model - for node in self.__complete_subgraph(self.sink_node) + for node in self.__all_nodes_in_subgraph(self.sink_node) if node._input_semantic_model is not None ] )