From 4686c911577d9e7b6c0a946668fbddc8916b67f5 Mon Sep 17 00:00:00 2001 From: Yee Hing Tong Date: Thu, 12 Mar 2026 12:12:25 -0700 Subject: [PATCH] add retry Signed-off-by: Yee Hing Tong --- flytekit/core/data_persistence.py | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/flytekit/core/data_persistence.py b/flytekit/core/data_persistence.py index 329bfe3ff8..74c4cf35c3 100644 --- a/flytekit/core/data_persistence.py +++ b/flytekit/core/data_persistence.py @@ -132,8 +132,9 @@ def retry_request(func, *args, **kwargs): sleep(random.randint(0, min(2**retry, 32))) return func(*args, **kwargs) except Exception as e: - # Catch this specific error message from S3 since S3FS doesn't catch it and retry the request. - if "Please reduce your request rate" in str(e): + # Catch specific retryable error messages from S3 since S3FS doesn't catch and retry them. + err_msg = str(e) + if "Please reduce your request rate" in err_msg or "RequestTimeTooSkewed" in err_msg: if retry == retries - 1: raise e else: