diff --git a/python/ray/data/context.py b/python/ray/data/context.py index 2280a18f9a446..3d9e745e148d6 100644 --- a/python/ray/data/context.py +++ b/python/ray/data/context.py @@ -102,6 +102,8 @@ DEFAULT_ENABLE_GET_OBJECT_LOCATIONS_FOR_METRICS = False +# `write_file_retry_on_errors` is deprecated in favor of `retried_io_errors`. You +# shouldn't need to modify `DEFAULT_WRITE_FILE_RETRY_ON_ERRORS`. DEFAULT_WRITE_FILE_RETRY_ON_ERRORS = ( "AWS Error INTERNAL_FAILURE", "AWS Error NETWORK_CONNECTION", @@ -114,6 +116,7 @@ "AWS Error NETWORK_CONNECTION", "AWS Error SLOW_DOWN", "AWS Error UNKNOWN (HTTP status 503)", + "AWS Error SERVICE_UNAVAILABLE", ) DEFAULT_WARN_ON_DRIVER_MEMORY_USAGE_BYTES = 2 * 1024 * 1024 * 1024