From af219e7e92a09159cf6041ac48a01db3ed1ea230 Mon Sep 17 00:00:00 2001 From: Judah Rand <17158624+judahrand@users.noreply.github.com> Date: Tue, 21 Sep 2021 14:33:17 +0100 Subject: [PATCH] Parse BQ `DATETIME` and `TIMESTAMP` (#1885) Signed-off-by: Judah Rand <17158624+judahrand@users.noreply.github.com> --- sdk/python/feast/type_map.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/sdk/python/feast/type_map.py b/sdk/python/feast/type_map.py index 7646ad0c90..80d744544d 100644 --- 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,