fix: Snowflake infer snowflake OBJECT return type for number inference

Signed-off-by: miles.adkins <miles.adkins@snowflake.com>
feast-dev · Oct 26, 2022 · 3b3fbac · 3b3fbac
1 parent 1935294
commit 3b3fbac
Showing 1 changed file with 7 additions and 2 deletions.
diff --git a/sdk/python/feast/infra/offline_stores/snowflake_source.py b/sdk/python/feast/infra/offline_stores/snowflake_source.py
@@ -265,8 +265,13 @@ def get_table_column_names_and_types(
                         else:
                             if len(result) > 0:
                                 max_value = result.iloc[0][0]
-                                if max_value is not None and len(str(max_value)) <= 18:
-                                    row["snowflake_type"] = "NUMBER"
+                                if max_value is not None and len(str(max_value)) <= 9:
+                                    row["snowflake_type"] = "NUMBER32"
+                                    continue
+                                elif (
+                                    max_value is not None and len(str(max_value)) <= 18
+                                ):
+                                    row["snowflake_type"] = "NUMBER64"
                                     continue
                             raise NotImplementedError(
                                 "NaNs or Numbers larger than INT64 are not supported"