Skip to content

Commit 71d16b3

Browse files
authored
Merge branch 'master' into fix-issue-2426
2 parents b0dd7a1 + 8ad82fe commit 71d16b3

File tree

1 file changed

+5
-0
lines changed

1 file changed

+5
-0
lines changed

src/sagemaker/feature_store/feature_group.py

Lines changed: 5 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -21,6 +21,7 @@
2121

2222
from __future__ import absolute_import
2323

24+
import copy
2425
import logging
2526
import math
2627
import os
@@ -193,6 +194,10 @@ def _ingest_single_batch(
193194
Returns:
194195
List of row indices that failed to be ingested.
195196
"""
197+
retry_config = client_config.retries
198+
if "max_attempts" not in retry_config and "total_max_attempts" not in retry_config:
199+
client_config = copy.deepcopy(client_config)
200+
client_config.retries = {"max_attempts": 10, "mode": "standard"}
196201
sagemaker_featurestore_runtime_client = boto3.Session().client(
197202
service_name="sagemaker-featurestore-runtime", config=client_config
198203
)

0 commit comments

Comments
 (0)