Fix imports.

metricflow/sql/sql_cte_node.py

@@ -2,15 +2,16 @@
 
 from dataclasses import dataclass
 from functools import cached_property
-from typing import override, Optional, Sequence, Tuple, Mapping
+from typing import Mapping, Optional, Sequence, Tuple, override
 
-from metricflow.sql.sql_plan import SqlPlanNode, SqlPlanNodeVisitor, SqlSelectColumn
-from metricflow.sql.sql_select_node import SqlSelectStatementNode
-from metricflow.sql.sql_table_node import SqlTableNode
 from metricflow_semantics.collection_helpers.merger import Mergeable
 from metricflow_semantics.dag.id_prefix import IdPrefix, StaticIdPrefix
 from metricflow_semantics.visitor import VisitorOutputT
 
+from metricflow.sql.sql_plan import SqlPlanNode, SqlPlanNodeVisitor, SqlSelectColumn
+from metricflow.sql.sql_select_node import SqlSelectStatementNode
+from metricflow.sql.sql_table_node import SqlTableNode
+
 
 @dataclass(frozen=True, eq=False)
 class SqlCteNode(SqlPlanNode):

metricflow/sql/sql_plan.py

@@ -3,6 +3,7 @@
 from __future__ import annotations
 
 import logging
+import typing
 from abc import ABC, abstractmethod
 from dataclasses import dataclass
 from typing import Generic, Optional, Sequence
@@ -13,11 +14,12 @@
 from metricflow_semantics.visitor import VisitorOutputT
 from typing_extensions import Self
 
-from metricflow.sql.sql_ctas_node import SqlCreateTableAsNode
-from metricflow.sql.sql_cte_node import SqlCteAliasMapping, SqlCteNode
-from metricflow.sql.sql_select_node import SqlSelectStatementNode
-from metricflow.sql.sql_select_text_node import SqlSelectTextNode
-from metricflow.sql.sql_table_node import SqlTableNode
+if typing.TYPE_CHECKING:
+    from metricflow.sql.sql_ctas_node import SqlCreateTableAsNode
+    from metricflow.sql.sql_cte_node import SqlCteAliasMapping, SqlCteNode
+    from metricflow.sql.sql_select_node import SqlSelectStatementNode
+    from metricflow.sql.sql_select_text_node import SqlSelectTextNode
+    from metricflow.sql.sql_table_node import SqlTableNode
 
 logger = logging.getLogger(__name__)
 

metricflow/sql/sql_select_node.py

@@ -1,5 +1,6 @@
 from __future__ import annotations
 
+import typing
 from dataclasses import dataclass
 from typing import Optional, Sequence, Tuple, override
 
@@ -9,10 +10,12 @@
 from metricflow_semantics.sql.sql_join_type import SqlJoinType
 from metricflow_semantics.visitor import VisitorOutputT
 
-from metricflow.sql.sql_cte_node import SqlCteAliasMapping, SqlCteNode
 from metricflow.sql.sql_plan import SqlPlanNode, SqlPlanNodeVisitor, SqlSelectColumn
 from metricflow.sql.sql_table_node import SqlTableNode
 
+if typing.TYPE_CHECKING:
+    from metricflow.sql.sql_cte_node import SqlCteAliasMapping, SqlCteNode
+
 
 @dataclass(frozen=True)
 class SqlJoinDescription:

metricflow/sql/sql_table_node.py

@@ -1,5 +1,6 @@
 from __future__ import annotations
 
+import typing
 from dataclasses import dataclass
 from typing import Optional, Sequence, override
 
@@ -8,9 +9,11 @@
 from metricflow_semantics.sql.sql_table import SqlTable
 from metricflow_semantics.visitor import VisitorOutputT
 
-from metricflow.sql.sql_cte_node import SqlCteAliasMapping
 from metricflow.sql.sql_plan import SqlPlanNode, SqlPlanNodeVisitor, SqlSelectColumn
-from metricflow.sql.sql_select_node import SqlSelectStatementNode
+
+if typing.TYPE_CHECKING:
+    from metricflow.sql.sql_cte_node import SqlCteAliasMapping
+    from metricflow.sql.sql_select_node import SqlSelectStatementNode
 
 
 @dataclass(frozen=True, eq=False)