Merge pull request #4 from robhowley/rh-int-float

allow for ints to go to float features
feast-dev · Jan 12, 2024 · 7af112b · 7af112b
2 parents e0ead80 + 4769b79
commit 7af112b
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/sdk/python/feast/type_map.py b/sdk/python/feast/type_map.py
@@ -298,7 +298,7 @@ def _type_err(item, dtype):
         None,
     ),
     ValueType.FLOAT: ("float_val", lambda x: float(x), None),
-    ValueType.DOUBLE: ("double_val", lambda x: x, {float, np.float64}),
+    ValueType.DOUBLE: ("double_val", lambda x: x, {float, np.float64, int, np.int_}),
     ValueType.STRING: ("string_val", lambda x: str(x), None),
     ValueType.BYTES: ("bytes_val", lambda x: x, {bytes}),
     ValueType.BOOL: ("bool_val", lambda x: x, {bool, np.bool_, int, np.int_}),