Parse BQ DATETIME and TIMESTAMP

Signed-off-by: Judah Rand <17158624+judahrand@users.noreply.github.com>
feast-dev · Sep 20, 2021 · 812efe4 · 812efe4
1 parent 75f67ec
commit 812efe4
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/sdk/python/feast/type_map.py b/sdk/python/feast/type_map.py
@@ -340,8 +340,8 @@ def pa_to_feast_value_type(pa_type_as_str: str) -> ValueType:
 
 def bq_to_feast_value_type(bq_type_as_str):
     type_map: Dict[ValueType, Union[str, Dict[str, Any]]] = {
-        "DATETIME": ValueType.STRING,  # Update to ValueType.UNIX_TIMESTAMP once #1520 lands.
-        "TIMESTAMP": ValueType.STRING,  # Update to ValueType.UNIX_TIMESTAMP once #1520 lands.
+        "DATETIME": ValueType.UNIX_TIMESTAMP,
+        "TIMESTAMP": ValueType.UNIX_TIMESTAMP,
         "INTEGER": ValueType.INT64,
         "INT64": ValueType.INT64,
         "STRING": ValueType.STRING,