diff --git a/src/loader.py b/src/loader.py index 9e3dcce..45b97f3 100644 --- a/src/loader.py +++ b/src/loader.py @@ -22,7 +22,8 @@ class EtlLoader: self.etl = EtlConfig() self.api_config = ApiConfig() - @backoff.on_exception(backoff.expo, (asyncio.TimeoutError, aiohttp.ClientConnectorError,), base=2, factor=1, + @backoff.on_exception(backoff.expo, (asyncio.TimeoutError, aiohttp.ClientConnectorError, + aiohttp.ServerDisconnectedError), base=2, factor=1, max_value=int(misc_settings.max_wait_size), max_tries=None) async def load_data(self, state: State, abitr: Abitr, iblock_id: int, storage: BaseStorage): async with aiohttp.ClientSession() as session: