Namespace dataflow plan subgraphs by prefix

dbt-labs · May 22, 2024 · 9ef3244 · 9ef3244
1 parent fda2d1b
commit 9ef3244
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 1 deletion.
diff --git a/metricflow-semantics/metricflow_semantics/dag/id_prefix.py b/metricflow-semantics/metricflow_semantics/dag/id_prefix.py
@@ -89,6 +89,7 @@ class StaticIdPrefix(IdPrefix, Enum, metaclass=EnumMetaClassHelper):
     VALUES_GROUP_BY_ITEM_RESOLUTION_NODE = "vr"
 
     DATAFLOW_PLAN_PREFIX = "dfp"
+    DATAFLOW_PLAN_SUBGRAPH_PREFIX = "dfpsub"
     OPTIMIZED_DATAFLOW_PLAN_PREFIX = "dfpo"
     SQL_QUERY_PLAN_PREFIX = "sqp"
     EXEC_PLAN_PREFIX = "ep"

diff --git a/metricflow/dataflow/dataflow_plan.py b/metricflow/dataflow/dataflow_plan.py
@@ -78,7 +78,9 @@ def as_plan(self) -> DataflowPlan:
         inside of each node, we make those properties of the DataflowPlan, and this node-level converter makes
         such properties easily accessible.
         """
-        return DataflowPlan(sink_nodes=(self,))
+        return DataflowPlan(
+            sink_nodes=(self,), plan_id=DagId.from_id_prefix(id_prefix=StaticIdPrefix.DATAFLOW_PLAN_SUBGRAPH_PREFIX)
+        )
 
     @abstractmethod
     def accept(self, visitor: DataflowPlanNodeVisitor[VisitorOutputT]) -> VisitorOutputT: