From 72bae7d39e3f08df3f98743c412f408924ac3261 Mon Sep 17 00:00:00 2001 From: ZF Deng Date: Fri, 7 Jan 2022 22:07:25 -0800 Subject: [PATCH 1/2] fix datetime64 to UNIX_TS --- sdk/python/feast/type_map.py | 4 ++++ 1 file changed, 4 insertions(+) diff --git a/sdk/python/feast/type_map.py b/sdk/python/feast/type_map.py index 969ca65862..80eddc323c 100644 --- a/sdk/python/feast/type_map.py +++ b/sdk/python/feast/type_map.py @@ -290,6 +290,10 @@ def _python_value_to_proto_value( return [ ProtoValue(int64_val=int(value.ToSeconds())) for value in values ] + elif isinstance(sample, np.datetime64): + return [ + ProtoValue(int64_val=int(pd.Timestamp(value).to_pydatetime().timestamp())) for value in values + ] return [ProtoValue(int64_val=int(value)) for value in values] if feast_value_type in PYTHON_SCALAR_VALUE_TYPE_TO_PROTO_VALUE: From f47840d824f4db1da77490f6090dfe42bbdc3b29 Mon Sep 17 00:00:00 2001 From: ZF Deng Date: Fri, 7 Jan 2022 23:07:46 -0800 Subject: [PATCH 2/2] old version of numpy use bool_ --- sdk/python/feast/type_map.py | 1 + 1 file changed, 1 insertion(+) diff --git a/sdk/python/feast/type_map.py b/sdk/python/feast/type_map.py index 80eddc323c..46c6b3c39a 100644 --- a/sdk/python/feast/type_map.py +++ b/sdk/python/feast/type_map.py @@ -108,6 +108,7 @@ def python_type_to_feast_value_type( "uint8": ValueType.INT32, "int8": ValueType.INT32, "bool": ValueType.BOOL, + "bool_": ValueType.BOOL "timedelta": ValueType.UNIX_TIMESTAMP, "timestamp": ValueType.UNIX_TIMESTAMP, "datetime": ValueType.UNIX_TIMESTAMP,