fix: Fixed data mapping errors for Snowflake (#2558)

sfc-gh-madkins · web-flow · commit 53c2ce2c2255 · 2022-04-15T21:26:24.000-07:00
* fixed data mapping errors

Signed-off-by: Miles Adkins &lt;miles.adkins@snowflake.com&gt;

* fixed data mapping errors

Signed-off-by: Miles Adkins &lt;miles.adkins@snowflake.com&gt;
diff --git a/sdk/python/feast/type_map.py b/sdk/python/feast/type_map.py
@@ -529,7 +529,8 @@ def snowflake_python_type_to_feast_value_type(
         "uint8": ValueType.INT32,
         "int8": ValueType.INT32,
         "datetime64[ns]": ValueType.UNIX_TIMESTAMP,
-        "object": ValueType.UNKNOWN,
+        "object": ValueType.STRING,
+        "bool": ValueType.BOOL,
     }
 
     return type_map[snowflake_python_type_as_str.lower()]

Original file line number	Diff line number	Diff line change
`@@ -529,7 +529,8 @@ def snowflake_python_type_to_feast_value_type(`
`529`	`529`	`"uint8": ValueType.INT32,`
`530`	`530`	`"int8": ValueType.INT32,`
`531`	`531`	`"datetime64[ns]": ValueType.UNIX_TIMESTAMP,`
`532`		`- "object": ValueType.UNKNOWN,`
	`532`	`+ "object": ValueType.STRING,`
	`533`	`+ "bool": ValueType.BOOL,`
`533`	`534`	`}`
`534`	`535`
`535`	`536`	`return type_map[snowflake_python_type_as_str.lower()]`