diff --git a/src/sagemaker/feature_store/feature_group.py b/src/sagemaker/feature_store/feature_group.py index 689b70c0d7..7cce7a2dca 100644 --- 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(