diff --git a/src/databricks/sqlalchemy/_ddl.py b/src/databricks/sqlalchemy/_ddl.py index d5d0bf8..1be86c3 100644 --- a/src/databricks/sqlalchemy/_ddl.py +++ b/src/databricks/sqlalchemy/_ddl.py @@ -98,3 +98,9 @@ def limit_clause(self, select, **kw): text += "\n LIMIT ALL" text += " OFFSET " + self.process(select._offset_clause, **kw) return text + + def visit_try_cast(self, element, **kw): + return "try_cast(%s AS %s)" % ( + self.process(element.clause, **kw), + self.process(element.typeclause, **kw), + )