Fixed annotations download

Author: VaghinakDev

src/superannotate/lib/core/usecases/annotations.py

@@ -1386,11 +1386,11 @@ async def download_small_annotations(
         )
 
     async def distribute_to_queues(
-        self, item_names, sm_queue_id, l_queue_id, folder_id
+        self, item_names, sm_queue_id, l_queue_id, folder: FolderEntity
     ):
         try:
             resp = self._service_provider.annotations.sort_items_by_size(
-                project=self._project, folder=self._folder, item_names=item_names
+                project=self._project, folder=folder, item_names=item_names
             )
 
             for item in resp["large"]:
@@ -1410,7 +1410,7 @@ async def run_workers(self, item_names, folder: FolderEntity, export_path):
             big_file_queue_idx = len(self._big_file_queues) - 1
             res = await asyncio.gather(
                 self.distribute_to_queues(
-                    item_names, small_file_queue_idx, big_file_queue_idx, folder.id
+                    item_names, small_file_queue_idx, big_file_queue_idx, folder
                 ),
                 self.download_big_annotations(big_file_queue_idx, export_path),
                 self.download_big_annotations(big_file_queue_idx, export_path),

src/superannotate/lib/infrastructure/services/http_client.py

@@ -16,6 +16,8 @@
 from lib.core.exceptions import AppException
 from lib.core.service_types import ServiceResponse
 from lib.core.serviceproviders import BaseClient
+from requests.adapters import HTTPAdapter
+from requests.adapters import Retry
 from superannotate import __version__
 
 logger = logging.getLogger("sa")
@@ -39,7 +41,11 @@ def __init__(self, api_url: str, token: str, verify_ssl: bool = True):
     def _get_session(self, thread_id, ttl=None):  # noqa
         del ttl
         del thread_id
+        retries = Retry(total=3, backoff_factor=0.1, status_forcelist=[502, 503, 504])
+
         session = requests.Session()
+        session.mount("http://", HTTPAdapter(max_retries=retries))  # noqa
+        session.mount("https://", HTTPAdapter(max_retries=retries))
         session.headers.update(self.default_headers)
         return session
 

tests/integration/annotations/test_upload_annotations_from_folder_to_project.py

@@ -2,8 +2,8 @@
 from pathlib import Path
 
 from src.superannotate import SAClient
-from tests.integration.base import BaseTestCase
 from tests import DATA_SET_PATH
+from tests.integration.base import BaseTestCase
 
 sa = SAClient()
 
@@ -33,7 +33,7 @@ def big_annotations_folder_path(self):
     @property
     def large_annotations_folder_path(self):
         return os.path.join(DATA_SET_PATH, self.TEST_LARGE_FOLDER_PATH)
-    
+
     def test_annotation_folder_upload_download(self):
         self._attach_items()
         sa.create_annotation_classes_from_classes_json(