ins-pricing 0.1.7__tar.gz → 0.1.8__tar.gz

{ins_pricing-0.1.7 → ins_pricing-0.1.8}/Ins_Pricing.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: ins_pricing
-Version: 0.1.7
+Version: 0.1.8
 Summary: Reusable modelling, pricing, governance, and reporting utilities.
 Author: meishi125478
 License: Proprietary

{ins_pricing-0.1.7 → ins_pricing-0.1.8}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: ins_pricing
-Version: 0.1.7
+Version: 0.1.8
 Summary: Reusable modelling, pricing, governance, and reporting utilities.
 Author: meishi125478
 License: Proprietary

{ins_pricing-0.1.7 → ins_pricing-0.1.8}/ins_pricing/modelling/BayesOpt_entry.py

@@ -359,6 +359,24 @@ def train_from_config(args: argparse.Namespace) -> None:
     dist_world_size = _safe_int_env("WORLD_SIZE", 1)
     dist_rank = _safe_int_env("RANK", 0)
     dist_active = dist_world_size > 1
+    def _ddp_barrier(reason: str) -> None:
+        if not dist_active:
+            return
+        torch_mod = getattr(ropt, "torch", None)
+        dist_mod = getattr(torch_mod, "distributed", None)
+        if dist_mod is None:
+            return
+        try:
+            if not getattr(dist_mod, "is_available", lambda: False)():
+                return
+            if not dist_mod.is_initialized():
+                ddp_ok, _, _, _ = ropt.DistributedUtils.setup_ddp()
+                if not ddp_ok or not dist_mod.is_initialized():
+                    return
+            dist_mod.barrier()
+        except Exception as exc:
+            print(f"[DDP] barrier failed during {reason}: {exc}", flush=True)
+            raise
 
     data_dir = Path(cfg["data_dir"])
     data_dir.mkdir(parents=True, exist_ok=True)
@@ -536,6 +554,7 @@ def train_from_config(args: argparse.Namespace) -> None:
                     print(
                         f"[Rank {dist_rank}] Skip {model_name}/{key} because trainer is not DDP-enabled."
                     )
+                    _ddp_barrier(f"skip_non_ddp_{model_name}_{key}")
                     continue
 
             print(
@@ -545,6 +564,8 @@ def train_from_config(args: argparse.Namespace) -> None:
             _plot_loss_curve_for_trainer(model_name, model.trainers[key])
             if key in PYTORCH_TRAINERS:
                 ropt.free_cuda()
+            if dist_active and not trainer_uses_ddp:
+                _ddp_barrier(f"finish_non_ddp_{model_name}_{key}")
             executed_keys.append(key)
 
         if not executed_keys:

{ins_pricing-0.1.7 → ins_pricing-0.1.8}/ins_pricing/setup.py

@@ -14,7 +14,7 @@ def _discover_packages() -> list[str]:
 
 setup(
     name="Ins-Pricing",
-    version="0.1.7",
+    version="0.1.8",
     description="Reusable modelling, pricing, governance, and reporting utilities.",
     author="meishi125478",
     license="Proprietary",

{ins_pricing-0.1.7 → ins_pricing-0.1.8}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "ins_pricing"
-version = "0.1.7"
+version = "0.1.8"
 description = "Reusable modelling, pricing, governance, and reporting utilities."
 readme = "README.md"
 requires-python = ">=3.9"