We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
2 parents b0dd7a1 + 8ad82fe commit 71d16b3Copy full SHA for 71d16b3
src/sagemaker/feature_store/feature_group.py
@@ -21,6 +21,7 @@
21
22
from __future__ import absolute_import
23
24
+import copy
25
import logging
26
import math
27
import os
@@ -193,6 +194,10 @@ def _ingest_single_batch(
193
194
Returns:
195
List of row indices that failed to be ingested.
196
"""
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"}
201
sagemaker_featurestore_runtime_client = boto3.Session().client(
202
service_name="sagemaker-featurestore-runtime", config=client_config
203
)
0 commit comments