Array testing 2

PeterDKay · PeterDKay · commit 2f9a40aa3de3 · 2024-01-05T18:53:14.000Z
diff --git a/src/databricks/sqlalchemy/dialect/__init__.py b/src/databricks/sqlalchemy/dialect/__init__.py
@@ -26,15 +26,9 @@
     from alembic.ddl import DefaultImpl
     from alembic.autogenerate import render
 
-    # TODO:
     class DatabricksImpl(DefaultImpl):
         __dialect__ = "databricks"
 
-        def _render_ARRAY_type(self, type_, autogen_context):
-            return render._render_type_w_subtype(
-                type_, autogen_context, "item_type", r"(.+?\()"
-            )
-
 
 class DatabricksDecimal(types.TypeDecorator):
     """Translates strings to decimals"""
diff --git a/src/databricks/sqlalchemy/dialect/compiler.py b/src/databricks/sqlalchemy/dialect/compiler.py
@@ -44,3 +44,6 @@ def visit_DATETIME(self, type_):
 
     def visit_JSON(self, type_):
         return "STRUCT"
+
+    def visit_ARRAY(self, type_):
+        return "ARRAY {type}".format(type=type_.item_type)
