archeo 2.0.0.dev4__tar.gz → 2.0.0.dev5__tar.gz

{archeo-2.0.0.dev4 → archeo-2.0.0.dev5}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: archeo
-Version: 2.0.0.dev4
+Version: 2.0.0.dev5
 Summary: Bayesian framework for inferring natal kick, ancestral masses and spins of black holes.
 Author-email: wyhwong <wyhwong@link.cuhk.edu.hk>
 License-Expression: MIT
@@ -32,6 +32,7 @@ Requires-Dist: scipy>=1.17.1
 Requires-Dist: seaborn>=0.13.2
 Requires-Dist: surfinbh>=1.2.7
 Requires-Dist: tabulate>=0.10.0
+Requires-Dist: tenacity>=9.1.4
 Requires-Dist: tqdm>=4.67.3
 Dynamic: license-file
 

{archeo-2.0.0.dev4 → archeo-2.0.0.dev5}/archeo/constants/enum.py

@@ -1,6 +1,8 @@
 import enum
 import os
 
+from tenacity import retry, retry_if_exception_type, stop_after_attempt, wait_fixed
+
 from archeo.utils.logger import get_logger
 
 
@@ -22,6 +24,11 @@ class Fits(enum.StrEnum):
     NRSUR7DQ4REMNANT = "NRSur7dq4Remnant"
     SURFINBH7DQ2 = "surfinBH7dq2"
 
+    @retry(
+        stop=stop_after_attempt(3),
+        wait=wait_fixed(3),
+        retry=retry_if_exception_type(RuntimeError),
+    )
     def load(self):
         """Load a surfinBH fits.
 
@@ -42,6 +49,9 @@ class Fits(enum.StrEnum):
         )
 
         try:
+            # NOTE: We added a retry logic because the model loading can fail
+            # due to IO limitation of h5py. This happens when we have multiple
+            # processes trying to load the same model at the same time.
             return surfinBH.LoadFits(self.value)
         except (OSError, KeyError) as e:
             LOGGER.error("Failed to load surfinBH %s: %s", self.value, str(e))

{archeo-2.0.0.dev4 → archeo-2.0.0.dev5}/archeo.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: archeo
-Version: 2.0.0.dev4
+Version: 2.0.0.dev5
 Summary: Bayesian framework for inferring natal kick, ancestral masses and spins of black holes.
 Author-email: wyhwong <wyhwong@link.cuhk.edu.hk>
 License-Expression: MIT
@@ -32,6 +32,7 @@ Requires-Dist: scipy>=1.17.1
 Requires-Dist: seaborn>=0.13.2
 Requires-Dist: surfinbh>=1.2.7
 Requires-Dist: tabulate>=0.10.0
+Requires-Dist: tenacity>=9.1.4
 Requires-Dist: tqdm>=4.67.3
 Dynamic: license-file
 

{archeo-2.0.0.dev4 → archeo-2.0.0.dev5}/archeo.egg-info/requires.txt

@@ -10,4 +10,5 @@ scipy>=1.17.1
 seaborn>=0.13.2
 surfinbh>=1.2.7
 tabulate>=0.10.0
+tenacity>=9.1.4
 tqdm>=4.67.3

{archeo-2.0.0.dev4 → archeo-2.0.0.dev5}/pyproject.toml

@@ -16,7 +16,7 @@ classifiers = [
     "Operating System :: OS Independent"
 ]
 keywords = ["black-holes", "gravitational-waves", "black-hole-archeology"]
-version = "2.0.0.dev4"
+version = "2.0.0.dev5"
 readme = "README.md"
 requires-python = ">=3.11,<3.14"
 dependencies = [
@@ -32,6 +32,7 @@ dependencies = [
     "seaborn>=0.13.2",
     "surfinbh>=1.2.7",
     "tabulate>=0.10.0",
+    "tenacity>=9.1.4",
     "tqdm>=4.67.3",
 ]