Merge pull request #42 from sede-open/hotfix/array-string-type

PeterDKay · web-flow · commit 543b34cbd5e4 · 2024-01-08T15:11:34.000Z
Hotfix/array string type
diff --git a/src/databricks/sqlalchemy/dialect/compiler.py b/src/databricks/sqlalchemy/dialect/compiler.py
@@ -46,7 +46,9 @@ def visit_JSON(self, type_):
         return "STRUCT"
 
     def visit_ARRAY(self, type_):
-        if type_.item_type == 'VARCHAR':
+        element_type = type_.item_type
+        element_type = str(element_type)
+        if element_type == 'VARCHAR':
             return "ARRAY<STRING>"
         else:
             return "ARRAY<{type}>".format(type=type_.item_type)
