Handle np.datetime64 to ValueType.UNIX_TIMESTAMP conversion

Signed-off-by: Judah Rand <[email protected]>
feast-dev · Jan 21, 2022 · 77af10f · 77af10f
1 parent 62fae05
commit 77af10f
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/sdk/python/feast/type_map.py b/sdk/python/feast/type_map.py
@@ -290,6 +290,11 @@ 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=value.astype("datetime64[s]").astype("int"))
+                    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: