ins-pricing 0.1.7__py3-none-any.whl → 0.1.8__py3-none-any.whl

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/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.dist-info → ins_pricing-0.1.8.dist-info}/METADATA

@@ -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.dist-info → ins_pricing-0.1.8.dist-info}/RECORD

@@ -1,6 +1,6 @@
 ins_pricing/README.md,sha256=vufPPg719SZzrtcplnIN5BJvBsXKzIgxCUeCrWUzobM,2495
 ins_pricing/__init__.py,sha256=hhZymgqtvilNC4WrZFXuT3-o0INSoVuRR-HA5EqyGAY,3085
-ins_pricing/setup.py,sha256=pTpRa5J64ssxmzVs0SLvLkFIawt9R48MfpgbTbjalKs,2034
+ins_pricing/setup.py,sha256=YHqqUtjbg2KrLkvXV_d_uGOCjzj9GUJaeCaR2omIknE,2034
 ins_pricing/governance/README.md,sha256=XnXLS5RPzWhEiicJ3WtGmpN935jppHhPftA9Lo2DPnQ,511
 ins_pricing/governance/__init__.py,sha256=d8tiDhOvHvAvgSohY1xv0vuEeHj8Gl1apQtw7ryEKM0,517
 ins_pricing/governance/approval.py,sha256=cjJQjU1ziR-d-9wVSXyMyX6S5zijJqDWERZNxjqGAUE,2879
@@ -9,7 +9,7 @@ ins_pricing/governance/registry.py,sha256=2uxQL6qMGY5IYWJti9MpaV_auvL--piJaasFrX
 ins_pricing/governance/release.py,sha256=ltyFIdeKbwj9fnEDxcQCURaQ5Zc_j0mqXFPNunmX_NQ,4743
 ins_pricing/modelling/BayesOpt.py,sha256=i2tB3c6EeucjKAsHyicGDNU7DVVCTihg-TgSoM1y18E,3332
 ins_pricing/modelling/BayesOpt_USAGE.md,sha256=x_CXot3RvGRYtt2xU2lrEfmtTbh2KGTryC3BJEZaoBw,39333
-ins_pricing/modelling/BayesOpt_entry.py,sha256=55odKaWpnFJzdNuEGzZMkvqawgsb-WVgc8Y8464MscY,21577
+ins_pricing/modelling/BayesOpt_entry.py,sha256=DMr0EjMA7YsawOA6ihrHMgJ6ZO9HmmfVsM9PJ-zzm2s,22484
 ins_pricing/modelling/BayesOpt_incremental.py,sha256=yE3qTXV63c8Jy8aKlY0vtx-Grv5uUbN-tuWRBk_osPE,28564
 ins_pricing/modelling/Explain_Run.py,sha256=y-OKxmT475iRtDBukXh7jwiBiRmjh7ywCVhRFKk4EZw,962
 ins_pricing/modelling/Explain_entry.py,sha256=e7EviU2fIWwyBsmoTq2ZM_rDKjZypQnFH68Q_rZVz1U,21330
@@ -163,7 +163,7 @@ user_packages legacy/Try/ModelBayesOptSearch.py,sha256=Co--dkbKxqpl-2tP9zEYmcEos
 user_packages legacy/Try/ResNetBayesOptSearch.py,sha256=mp38lBs-TE3lAc8t8o4He5y1iwGIgrlPerurh_xRmqs,10942
 user_packages legacy/Try/XgbBayesOptSearch.py,sha256=pWjlyT5FSK4zNnng0ZrpTIF5o-wBwuavxFA0LE4B4oE,5534
 user_packages legacy/Try/xgbbayesopt.py,sha256=Vow-wrxcwozbTDPKJcl3ZzYgShd-4oks5Kzs74kXUvA,22754
-ins_pricing-0.1.7.dist-info/METADATA,sha256=HUa-M89yxUMVQ_51zg65BEHAhuNjZ5G-IVTGmYCjf0I,3650
-ins_pricing-0.1.7.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-ins_pricing-0.1.7.dist-info/top_level.txt,sha256=CUUE4W0DghRtQubDfBbIr2Kcvhdu-rOXxDD8Atjm_-c,66
-ins_pricing-0.1.7.dist-info/RECORD,,
+ins_pricing-0.1.8.dist-info/METADATA,sha256=Mhx74nepsgksHz1xUQ4Al1rOPflNXSaOgqMKfT5QMkg,3650
+ins_pricing-0.1.8.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+ins_pricing-0.1.8.dist-info/top_level.txt,sha256=CUUE4W0DghRtQubDfBbIr2Kcvhdu-rOXxDD8Atjm_-c,66
+ins_pricing-0.1.8.dist-info/RECORD,,