Update type map to allow it to infer datetime.date objects

Signed-off-by: Jesse Whitehouse <[email protected]>
databricks · Sep 25, 2023 · 554623f · 554623f
1 parent 3333ca0
commit 554623f
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/src/databricks/sql/utils.py b/src/databricks/sql/utils.py
@@ -529,6 +529,7 @@ def infer_types(params: list[DbSqlParameter]):
         int: DbSqlType.INTEGER,
         float: DbSqlType.FLOAT,
         datetime.datetime: DbSqlType.TIMESTAMP,
+        datetime.date: DbSqlType.DATE,
         bool: DbSqlType.BOOLEAN,
     }
     newParams = copy.deepcopy(params)