fix: adding .lower() so that the new Pandas dtypes, which start with Caps (e.g. Int64), will match the type lookup

brifordwylie · brifordwylie · commit ab4be8d7e747 · 2023-03-21T13:00:43.000-06:00
diff --git a/src/sagemaker/feature_store/feature_group.py b/src/sagemaker/feature_store/feature_group.py
@@ -689,7 +689,7 @@ def load_feature_definitions(
         feature_definitions = []
         for column in data_frame:
             feature_type = self.DTYPE_TO_FEATURE_DEFINITION_CLS_MAP.get(
-                str(data_frame[column].dtype), None
+                str(data_frame[column].dtype).lower(), None
             )
             if feature_type:
                 feature_definitions.append(

Original file line number	Diff line number	Diff line change
`@@ -689,7 +689,7 @@ def load_feature_definitions(`
`689`	`689`	`feature_definitions = []`
`690`	`690`	`for column in data_frame:`
`691`	`691`	`feature_type = self.DTYPE_TO_FEATURE_DEFINITION_CLS_MAP.get(`
`692`		`- str(data_frame[column].dtype), None`
	`692`	`+ str(data_frame[column].dtype).lower(), None`
`693`	`693`	`)`
`694`	`694`	`if feature_type:`
`695`	`695`	`feature_definitions.append(`