Skip to content

Commit b326230

Browse files
committed
change: if latest_ingestion=True, event_time_feature_name in get_feature_group_as_dataframe must specified
1 parent aad7888 commit b326230

File tree

1 file changed

+4
-0
lines changed

1 file changed

+4
-0
lines changed

src/sagemaker/feature_group_utils.py

+4
Original file line numberDiff line numberDiff line change
@@ -115,6 +115,10 @@ def get_feature_group_as_dataframe(feature_group_name: str, athena_bucket: str,
115115
if event_time_feature_name is not None:
116116
query += str(f'AND {event_time_feature_name}=(SELECT MAX({event_time_feature_name}) FROM '
117117
+ f'"sagemaker_featurestore"."{feature_group_name}")')
118+
else:
119+
exc = Exception('Argument event_time_feature_name must be specified when using latest_ingestion=True.')
120+
logger.exception(exc)
121+
raise exc
118122
query += ';'
119123

120124
if session is not None:

0 commit comments

Comments
 (0)