Skip to content

Commit 6805c31

Browse files
authored
SNOW-2096114: Fix XML error message when sql simplifier is disabled (#3371)
1 parent 0621868 commit 6805c31

File tree

1 file changed

+7
-7
lines changed

1 file changed

+7
-7
lines changed

src/snowflake/snowpark/_internal/analyzer/snowflake_plan.py

Lines changed: 7 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -284,14 +284,14 @@ def add_single_quote(string: str) -> str:
284284
def search_read_file_node(
285285
node: Union[SnowflakePlan, Selectable]
286286
) -> Optional[ReadFileNode]:
287+
source_plan = (
288+
node.source_plan
289+
if isinstance(node, SnowflakePlan)
290+
else node.snowflake_plan.source_plan
291+
)
292+
if isinstance(source_plan, ReadFileNode):
293+
return source_plan
287294
for child in node.children_plan_nodes:
288-
source_plan = (
289-
child.source_plan
290-
if isinstance(child, SnowflakePlan)
291-
else child.snowflake_plan.source_plan
292-
)
293-
if isinstance(source_plan, ReadFileNode):
294-
return source_plan
295295
result = search_read_file_node(child)
296296
if result:
297297
return result

0 commit comments

Comments
 (0)