fix: adding .lower() so new Pandas dtypes will match the type lookup. (#3740)

brifordwylie · mufaddal-rohawala · web-flow · commit 79795af57f4d · 2023-05-09T14:39:27.000-07:00
Co-authored-by: Mufaddal Rohawala &lt;89424143+mufaddal-rohawala@users.noreply.github.com&gt;
diff --git a/src/sagemaker/feature_store/feature_group.py b/src/sagemaker/feature_store/feature_group.py
@@ -741,7 +741,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
`@@ -741,7 +741,7 @@ def load_feature_definitions(`
`741`	`741`	`feature_definitions = []`
`742`	`742`	`for column in data_frame:`
`743`	`743`	`feature_type = self.DTYPE_TO_FEATURE_DEFINITION_CLS_MAP.get(`
`744`		`- str(data_frame[column].dtype), None`
	`744`	`+ str(data_frame[column].dtype).lower(), None`
`745`	`745`	`)`
`746`	`746`	`if feature_type:`
`747`	`747`	`feature_definitions.append(`