Merge pull request datafold#853 from datafold/fix_mssql_date_type

teraamp · web-flow · commit 88dbfad3ec52 · 2024-01-15T23:56:50.000+11:00
Fix mssql, databricks date types
diff --git a/data_diff/databases/databricks.py b/data_diff/databases/databricks.py
@@ -5,6 +5,7 @@
 import attrs
 
 from data_diff.abcs.database_types import (
+    Date,
     Integer,
     Float,
     Decimal,
@@ -52,6 +53,7 @@ class Dialect(BaseDialect):
         # Timestamps
         "TIMESTAMP": Timestamp,
         "TIMESTAMP_NTZ": Timestamp,
+        "DATE": Date,
         # Text
         "STRING": Text,
         "VARCHAR": Text,
diff --git a/data_diff/databases/mssql.py b/data_diff/databases/mssql.py
@@ -13,6 +13,7 @@
 )
 from data_diff.abcs.database_types import (
     JSON,
+    Date,
     NumericType,
     Timestamp,
     TimestampTZ,
@@ -46,7 +47,7 @@ class Dialect(BaseDialect):
         "datetime": Timestamp,
         "datetime2": Timestamp,
         "smalldatetime": Timestamp,
-        "date": Timestamp,
+        "date": Date,
         # Numbers
         "float": Float,
         "real": Float,
