fix pandas dependency issue (#2537)

snowflakedb · Oct 30, 2024 · 714e767 · 714e767
1 parent fd86b5b
commit 714e767
Showing 1 changed file with 11 additions and 11 deletions.
diff --git a/src/snowflake/snowpark/dataframe.py b/src/snowflake/snowpark/dataframe.py
@@ -4522,27 +4522,27 @@ def map(
         for i, col_name in enumerate(output_column_names)
     ]
 
-    def wrap_result(result):
-        if vectorized:
+    if vectorized:
+
+        def wrap_result(result):
             if isinstance(result, pandas.DataFrame) or isinstance(result, tuple):
                 return result
             return (result,)
 
-        if isinstance(result, Row):
-            return tuple(result)
-        elif isinstance(result, tuple):
-            return result
-        else:
-            return (result,)
-
-    if vectorized:
-
         class _MapFunc:
             def process(self, pdf):
                 return wrap_result(func(pdf))
 
     else:
 
+        def wrap_result(result):
+            if isinstance(result, Row):
+                return tuple(result)
+            elif isinstance(result, tuple):
+                return result
+            else:
+                return (result,)
+
         class _MapFunc:
             def process(self, *argv):
                 input_args_to_row = Row(*df_columns)