File tree Expand file tree Collapse file tree 1 file changed +7
-7
lines changed
src/snowflake/snowpark/_internal/analyzer Expand file tree Collapse file tree 1 file changed +7
-7
lines changed Original file line number Diff line number Diff 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
You can’t perform that action at this time.
0 commit comments