Fix Column handling in arrays

svenkreiss · Sep 26, 2021 · b44adf6 · b44adf6
1 parent 78fc641
commit b44adf6
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/pysparkling/sql/expressions/arrays.py b/pysparkling/sql/expressions/arrays.py
@@ -119,7 +119,7 @@ def args(self):
         )
 
     def data_type(self, schema):
-        if not self.keys:
+        if not isinstance(self.keys, Column) and not self.keys:
             return MapType(keyType=NullType, valueType=NullType)
         return MapType(
             keyType=self.keys[0].data_type(schema),