From f95a4f3836e5337a4845afc34414ff00667a5a1f Mon Sep 17 00:00:00 2001
From: yanghua <yanghua1127@gmail.com>
Date: Thu, 12 Sep 2024 12:43:09 +0800
Subject: [PATCH] Infra: introduce retry func warpper

---
 tosfs/stability.py | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/tosfs/stability.py b/tosfs/stability.py
index f3ea7f4..e53dc6e 100644
--- a/tosfs/stability.py
+++ b/tosfs/stability.py
@@ -61,6 +61,8 @@
     StreamConsumedError,
     RetryError,
     InterruptedError,
+    ConnectionResetError,
+    ConnectionError,
 }
 
 MAX_RETRY_NUM = 20