pybiolib 1.2.218__tar.gz → 1.2.223__tar.gz

{pybiolib-1.2.218 → pybiolib-1.2.223}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: pybiolib
-Version: 1.2.218
+Version: 1.2.223
 Summary: BioLib Python Client
 Home-page: https://github.com/biolib
 License: MIT

{pybiolib-1.2.218 → pybiolib-1.2.223}/biolib/_internal/http_client.py

@@ -78,6 +78,7 @@ class HttpClient:
         retries: int = 5,
         timeout_in_seconds: Optional[int] = None,
         response_path: Optional[str] = None,
+        retry_on_http_500: Optional[bool] = False,
     ) -> HttpResponse:
         if not HttpClient.ssl_context:
             HttpClient.ssl_context = _create_ssl_context()
@@ -112,6 +113,9 @@ class HttpClient:
                 if error.code == 429:
                     logger_no_user_data.warning(f'HTTP {method} request failed with status 429 for "{url}"')
                     last_error = error
+                elif error.code == 500 and retry_on_http_500:
+                    logger_no_user_data.warning(f'HTTP {method} request failed with status 500 for "{url}"')
+                    last_error = error
                 elif error.code == 502:
                     logger_no_user_data.warning(f'HTTP {method} request failed with status 502 for "{url}"')
                     last_error = error

{pybiolib-1.2.218 → pybiolib-1.2.223}/biolib/compute_node/utils.py

@@ -1,6 +1,6 @@
-from enum import Enum
 import random
 import string
+from enum import Enum
 
 from biolib.biolib_logging import logger
 
@@ -29,7 +29,7 @@ def get_package_type(package):
         return 'StdoutAndStderr'
 
     else:
-        raise Exception(f"Unexpected package type {package_type}")
+        raise Exception(f'Unexpected package type {package_type}')
 
 
 class SystemExceptionCodes(Enum):
@@ -68,6 +68,8 @@ class SystemExceptionCodes(Enum):
     FAILED_TO_ALLOCATE_JOB_TO_COMPUTE_NODE = 32
     NO_MODULES_FOUND_ON_JOB = 33
     FAILED_TO_INITIALIZE_DOCKER_EXECUTOR = 34
+    COMPUTE_NODE_SHUTDOWN = 35
+    COMPUTE_NODE_SHUTDOWN_DUE_TO_HEALTH_CHECK_FAILURE = 36
 
 
 SystemExceptionCodeMap = {
@@ -106,6 +108,8 @@ SystemExceptionCodeMap = {
     32: 'Failed to allocate job to compute node',
     33: 'No modules found on job',
     34: 'Failed to initialize docker executor',
+    35: 'Compute node shutdown',
+    36: 'Compute node shutdown due to health check failure',
 }
 
 

{pybiolib-1.2.218 → pybiolib-1.2.223}/biolib/utils/__init__.py

@@ -93,6 +93,7 @@ def _download_chunk(input_tuple: DownloadChunkInputTuple) -> bytes:
             headers={'range': f'bytes={start}-{end}'},
             timeout_in_seconds=300,  # timeout after 5 min
             retries=20,
+            retry_on_http_500=True,
         )
     except Exception as exception:
         logger_no_user_data.exception("Hit error downloading chunk")

{pybiolib-1.2.218 → pybiolib-1.2.223}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "pybiolib"
-version = "1.2.218"
+version = "1.2.223"
 description = "BioLib Python Client"
 readme = "PYPI_README.md"
 license = "MIT"