From 51f5aba82fd51eacb548802db9179ab38058824f Mon Sep 17 00:00:00 2001
From: Paul Yang <github@ardorful.com>
Date: Thu, 23 Jan 2025 12:41:06 -0800
Subject: [PATCH] Fix imports.

---
 metricflow/sql/sql_cte_node.py    |  9 +++++----
 metricflow/sql/sql_plan.py        | 12 +++++++-----
 metricflow/sql/sql_select_node.py |  5 ++++-
 metricflow/sql/sql_table_node.py  |  7 +++++--
 4 files changed, 21 insertions(+), 12 deletions(-)

diff --git a/metricflow/sql/sql_cte_node.py b/metricflow/sql/sql_cte_node.py
index a14e4ed72e..2f08f86a71 100644
--- a/metricflow/sql/sql_cte_node.py
+++ b/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):
diff --git a/metricflow/sql/sql_plan.py b/metricflow/sql/sql_plan.py
index a38d8caf21..584b4d38b9 100644
--- a/metricflow/sql/sql_plan.py
+++ b/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__)
 
diff --git a/metricflow/sql/sql_select_node.py b/metricflow/sql/sql_select_node.py
index 93f7d89a34..71154c35f7 100644
--- a/metricflow/sql/sql_select_node.py
+++ b/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:
diff --git a/metricflow/sql/sql_table_node.py b/metricflow/sql/sql_table_node.py
index 12860f03c2..9a51a9c3f8 100644
--- a/metricflow/sql/sql_table_node.py
+++ b/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)