fix failure

snowflakedb · Sep 16, 2024 · a1bed4b · a1bed4b
1 parent 5357951
commit a1bed4b
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 3 deletions.
diff --git a/src/snowflake/snowpark/_internal/analyzer/expression.py b/src/snowflake/snowpark/_internal/analyzer/expression.py
@@ -67,7 +67,8 @@ def derive_dependent_columns_with_duplication(
     """
     result = []
     for exp in expressions:
-        result.extend(exp.dependent_column_names_with_duplication())
+        if exp is not None:
+            result.extend(exp.dependent_column_names_with_duplication())
     return result
 
 
@@ -326,7 +327,8 @@ def dependent_column_names(self) -> Optional[AbstractSet[str]]:
     def dependent_column_names_with_duplication(self) -> List[str]:
         return (
             []
-            if self._dependent_column_names == COLUMN_DEPENDENCY_ALL
+            if (self._dependent_column_names == COLUMN_DEPENDENCY_ALL)
+            or (self._dependent_column_names is None)
             else list(self._dependent_column_names)
         )
 

diff --git a/src/snowflake/snowpark/_internal/analyzer/grouping_set.py b/src/snowflake/snowpark/_internal/analyzer/grouping_set.py
@@ -49,7 +49,7 @@ def dependent_column_names(self) -> Optional[AbstractSet[str]]:
         flattened_args = [exp for sublist in self.args for exp in sublist]
         return derive_dependent_columns(*flattened_args)
 
-    def dependent_column_names_with_duplication(self) -> list[str]:
+    def dependent_column_names_with_duplication(self) -> List[str]:
         flattened_args = [exp for sublist in self.args for exp in sublist]
         return derive_dependent_columns_with_duplication(*flattened_args)