WIP

Author: ezvz

api/py/ai/chronon/cli/plan/physical_node.py

@@ -4,30 +4,20 @@
 from ai.chronon.api.common.ttypes import TableDependency
 from ai.chronon.api.ttypes import GroupBy, Join, Model, StagingQuery
 from ai.chronon.cli.compile.compile_context import CompiledObj
+from ai.chronon.orchestration.ttypes import PhysicalNode
 
 
-@dataclass
-class PhysicalNode:
-    name: str
-    node_type: str  # NodeType
-    conf: CompiledObj
-    conf_hash: str
-    table_dependencies: List[TableDependency]
-    output_columns: List[str]
-    output_table: str
+def from_group_by(cls, group_by: GroupBy) -> List[PhysicalNode]:
+    raise NotImplementedError("Method not yet implemented")
 
-    @classmethod
-    def from_group_by(cls, group_by: GroupBy) -> List["PhysicalNode"]:
-        raise NotImplementedError("Method not yet implemented")
 
-    @classmethod
-    def from_join(cls, join: Join) -> List["PhysicalNode"]:
-        raise NotImplementedError("Method not yet implemented")
+def from_join(cls, join: Join) -> List[PhysicalNode]:
+    raise NotImplementedError("Method not yet implemented")
 
-    @classmethod
-    def from_staging_query(cls, staging_query: StagingQuery) -> List["PhysicalNode"]:
-        raise NotImplementedError("Method not yet implemented")
 
-    @classmethod
-    def from_model(cls, model: Model) -> List["PhysicalNode"]:
-        raise NotImplementedError("Method not yet implemented")
+def from_staging_query(cls, staging_query: StagingQuery) -> List[PhysicalNode]:
+    raise NotImplementedError("Method not yet implemented")
+
+
+def from_model(cls, model: Model) -> List[PhysicalNode]:
+    raise NotImplementedError("Method not yet implemented")

api/thrift/orchestration.thrift

@@ -95,6 +95,17 @@ union PhysicalNodeType {
     5: TableNodeType tableNodeType
 }
 
+struct PhysicalNode {
+    1: required string name,
+    2: required PhysicalNodeType nodeType,
+    3: required LogicalNode logicalNode,
+    4: required string confHash,
+    5: required list<common.TableDependency> tableDependencies,
+    6: required list<string> outputColumns,
+    7: required string output_table
+}
+
+
 // ====================== End of physical node types ======================