From ab4be8d7e7474433710f25b8da03c742b87a47d6 Mon Sep 17 00:00:00 2001 From: Brian Wylie Date: Tue, 21 Mar 2023 13:00:43 -0600 Subject: [PATCH] fix: adding .lower() so that the new Pandas dtypes, which start with Caps (e.g. Int64), will match the type lookup --- src/sagemaker/feature_store/feature_group.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/sagemaker/feature_store/feature_group.py b/src/sagemaker/feature_store/feature_group.py index be3f6589a8..a6eae3129e 100644 --- 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(