From c7bf29730038590a0a24a1a8942fa9b6c5db5c0e Mon Sep 17 00:00:00 2001 From: yanghua Date: Mon, 18 Nov 2024 21:20:16 +0800 Subject: [PATCH] Make _get_bucket_type retryable --- tosfs/core.py | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/tosfs/core.py b/tosfs/core.py index 81b875d..80f25b7 100644 --- a/tosfs/core.py +++ b/tosfs/core.py @@ -2191,7 +2191,10 @@ def _split_path(self, path: str) -> Tuple[str, str, Optional[str]]: ) def _get_bucket_type(self, bucket: str) -> str: - bucket_type = self.tos_client._get_bucket_type(bucket) + bucket_type = retryable_func_executor( + lambda: self.tos_client._get_bucket_type(bucket), + max_retry_num=self.max_retry_num, + ) if not bucket_type: return TOS_BUCKET_TYPE_FNS