fix(datafusion): handle timestamp types

ibis-project · Apr 13, 2024 · f762aae · f762aae
1 parent 609c15f
commit f762aae
Showing 1 changed file with 15 additions and 0 deletions.
diff --git a/ibis/backends/sql/datatypes.py b/ibis/backends/sql/datatypes.py
@@ -427,8 +427,23 @@ class DataFusionType(PostgresType):
     unknown_type_strings = {
         "utf8": dt.string,
         "float64": dt.float64,
+        "float32": dt.float32,
     }
 
+    @classmethod
+    def _from_sqlglot_TIMESTAMP(cls, scale, tz) -> dt.Timestamp:
+        scales = {"SECOND": 0, "MILLISECOND": 3, "MICROSECOND": 6, "NANOSECOND": 9}
+
+        assert tz is not None
+
+        tz = tz.this.this
+
+        return dt.Timestamp(
+            scale=scales[scale.this.this],
+            timezone=None if tz == "NONE" else tz,
+            nullable=cls.default_nullable,
+        )
+
 
 class MySQLType(SqlglotType):
     dialect = "mysql"