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

{ins_pricing-0.1.6 → 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.6
+Version: 0.1.8
 Summary: Reusable modelling, pricing, governance, and reporting utilities.
 Author: meishi125478
 License: Proprietary
@@ -15,17 +15,40 @@ Requires-Python: >=3.9
 Description-Content-Type: text/markdown
 Requires-Dist: numpy>=1.20
 Requires-Dist: pandas>=1.4
-Requires-Dist: torch>=1.13
-Requires-Dist: optuna>=3.0
-Requires-Dist: xgboost>=1.6
-Requires-Dist: scikit-learn>=1.1
-Requires-Dist: statsmodels>=0.13
-Requires-Dist: joblib>=1.2
-Requires-Dist: matplotlib>=3.5
-Requires-Dist: shap>=0.41
-Requires-Dist: contextily>=1.3
-Requires-Dist: pynndescent>=0.5
-Requires-Dist: torch-geometric>=2.3
+Provides-Extra: bayesopt
+Requires-Dist: torch>=1.13; extra == "bayesopt"
+Requires-Dist: optuna>=3.0; extra == "bayesopt"
+Requires-Dist: xgboost>=1.6; extra == "bayesopt"
+Requires-Dist: scikit-learn>=1.1; extra == "bayesopt"
+Requires-Dist: statsmodels>=0.13; extra == "bayesopt"
+Requires-Dist: joblib>=1.2; extra == "bayesopt"
+Requires-Dist: matplotlib>=3.5; extra == "bayesopt"
+Provides-Extra: plotting
+Requires-Dist: matplotlib>=3.5; extra == "plotting"
+Requires-Dist: scikit-learn>=1.1; extra == "plotting"
+Provides-Extra: explain
+Requires-Dist: torch>=1.13; extra == "explain"
+Requires-Dist: shap>=0.41; extra == "explain"
+Requires-Dist: scikit-learn>=1.1; extra == "explain"
+Provides-Extra: geo
+Requires-Dist: contextily>=1.3; extra == "geo"
+Requires-Dist: matplotlib>=3.5; extra == "geo"
+Provides-Extra: gnn
+Requires-Dist: torch>=1.13; extra == "gnn"
+Requires-Dist: pynndescent>=0.5; extra == "gnn"
+Requires-Dist: torch-geometric>=2.3; extra == "gnn"
+Provides-Extra: all
+Requires-Dist: torch>=1.13; extra == "all"
+Requires-Dist: optuna>=3.0; extra == "all"
+Requires-Dist: xgboost>=1.6; extra == "all"
+Requires-Dist: scikit-learn>=1.1; extra == "all"
+Requires-Dist: statsmodels>=0.13; extra == "all"
+Requires-Dist: joblib>=1.2; extra == "all"
+Requires-Dist: matplotlib>=3.5; extra == "all"
+Requires-Dist: shap>=0.41; extra == "all"
+Requires-Dist: contextily>=1.3; extra == "all"
+Requires-Dist: pynndescent>=0.5; extra == "all"
+Requires-Dist: torch-geometric>=2.3; extra == "all"
 
 # Ins-Pricing Overview
 

{ins_pricing-0.1.6 → ins_pricing-0.1.8}/Ins_Pricing.egg-info/SOURCES.txt

@@ -9,69 +9,6 @@ Ins_Pricing.egg-info/top_level.txt
 ins_pricing/README.md
 ins_pricing/__init__.py
 ins_pricing/setup.py
-ins_pricing v2/__init__.py
-ins_pricing v2/setup.py
-ins_pricing v2/smoke_test.py
-ins_pricing v2/governance/__init__.py
-ins_pricing v2/governance/approval.py
-ins_pricing v2/governance/audit.py
-ins_pricing v2/governance/registry.py
-ins_pricing v2/governance/release.py
-ins_pricing v2/modelling/Explain_Run.py
-ins_pricing v2/modelling/Pricing_Run.py
-ins_pricing v2/modelling/__init__.py
-ins_pricing v2/modelling/cli_common.py
-ins_pricing v2/modelling/config.py
-ins_pricing v2/modelling/config_preprocess.py
-ins_pricing v2/modelling/core.py
-ins_pricing v2/modelling/data_container.py
-ins_pricing v2/modelling/features.py
-ins_pricing v2/modelling/model_manager.py
-ins_pricing v2/modelling/model_plotting.py
-ins_pricing v2/modelling/models.py
-ins_pricing v2/modelling/notebook_utils.py
-ins_pricing v2/modelling/run_logging.py
-ins_pricing v2/modelling/trainers.py
-ins_pricing v2/modelling/utils.py
-ins_pricing v2/modelling/watchdog_run.py
-ins_pricing v2/modelling/explain/__init__.py
-ins_pricing v2/modelling/explain/gradients.py
-ins_pricing v2/modelling/explain/metrics.py
-ins_pricing v2/modelling/explain/permutation.py
-ins_pricing v2/modelling/explain/shap_utils.py
-ins_pricing v2/modelling/plotting/__init__.py
-ins_pricing v2/modelling/plotting/common.py
-ins_pricing v2/modelling/plotting/curves.py
-ins_pricing v2/modelling/plotting/diagnostics.py
-ins_pricing v2/modelling/plotting/geo.py
-ins_pricing v2/modelling/plotting/importance.py
-ins_pricing v2/modelling/tests/conftest.py
-ins_pricing v2/modelling/tests/test_cross_val_generic.py
-ins_pricing v2/modelling/tests/test_distributed_utils.py
-ins_pricing v2/modelling/tests/test_explain.py
-ins_pricing v2/modelling/tests/test_geo_tokens_split.py
-ins_pricing v2/modelling/tests/test_graph_cache.py
-ins_pricing v2/modelling/tests/test_plotting.py
-ins_pricing v2/modelling/tests/test_plotting_library.py
-ins_pricing v2/modelling/tests/test_preprocessor.py
-ins_pricing v2/pricing/__init__.py
-ins_pricing v2/pricing/calibration.py
-ins_pricing v2/pricing/data_quality.py
-ins_pricing v2/pricing/exposure.py
-ins_pricing v2/pricing/factors.py
-ins_pricing v2/pricing/monitoring.py
-ins_pricing v2/pricing/rate_table.py
-ins_pricing v2/production/__init__.py
-ins_pricing v2/production/drift.py
-ins_pricing v2/production/monitoring.py
-ins_pricing v2/production/scoring.py
-ins_pricing v2/reporting/__init__.py
-ins_pricing v2/reporting/report_builder.py
-ins_pricing v2/reporting/scheduler.py
-ins_pricing v2/scripts/BayesOpt_incremental.py
-ins_pricing v2/scripts/Explain_entry.py
-ins_pricing v2/scripts/__init__.py
-ins_pricing v2/scripts/train.py
 ins_pricing.egg-info/PKG-INFO
 ins_pricing.egg-info/SOURCES.txt
 ins_pricing.egg-info/dependency_links.txt
@@ -138,6 +75,69 @@ ins_pricing/reporting/README.md
 ins_pricing/reporting/__init__.py
 ins_pricing/reporting/report_builder.py
 ins_pricing/reporting/scheduler.py
+ins_pricing_gemini/__init__.py
+ins_pricing_gemini/setup.py
+ins_pricing_gemini/smoke_test.py
+ins_pricing_gemini/governance/__init__.py
+ins_pricing_gemini/governance/approval.py
+ins_pricing_gemini/governance/audit.py
+ins_pricing_gemini/governance/registry.py
+ins_pricing_gemini/governance/release.py
+ins_pricing_gemini/modelling/Explain_Run.py
+ins_pricing_gemini/modelling/Pricing_Run.py
+ins_pricing_gemini/modelling/__init__.py
+ins_pricing_gemini/modelling/cli_common.py
+ins_pricing_gemini/modelling/config.py
+ins_pricing_gemini/modelling/config_preprocess.py
+ins_pricing_gemini/modelling/core.py
+ins_pricing_gemini/modelling/data_container.py
+ins_pricing_gemini/modelling/features.py
+ins_pricing_gemini/modelling/model_manager.py
+ins_pricing_gemini/modelling/model_plotting.py
+ins_pricing_gemini/modelling/models.py
+ins_pricing_gemini/modelling/notebook_utils.py
+ins_pricing_gemini/modelling/run_logging.py
+ins_pricing_gemini/modelling/trainers.py
+ins_pricing_gemini/modelling/utils.py
+ins_pricing_gemini/modelling/watchdog_run.py
+ins_pricing_gemini/modelling/explain/__init__.py
+ins_pricing_gemini/modelling/explain/gradients.py
+ins_pricing_gemini/modelling/explain/metrics.py
+ins_pricing_gemini/modelling/explain/permutation.py
+ins_pricing_gemini/modelling/explain/shap_utils.py
+ins_pricing_gemini/modelling/plotting/__init__.py
+ins_pricing_gemini/modelling/plotting/common.py
+ins_pricing_gemini/modelling/plotting/curves.py
+ins_pricing_gemini/modelling/plotting/diagnostics.py
+ins_pricing_gemini/modelling/plotting/geo.py
+ins_pricing_gemini/modelling/plotting/importance.py
+ins_pricing_gemini/modelling/tests/conftest.py
+ins_pricing_gemini/modelling/tests/test_cross_val_generic.py
+ins_pricing_gemini/modelling/tests/test_distributed_utils.py
+ins_pricing_gemini/modelling/tests/test_explain.py
+ins_pricing_gemini/modelling/tests/test_geo_tokens_split.py
+ins_pricing_gemini/modelling/tests/test_graph_cache.py
+ins_pricing_gemini/modelling/tests/test_plotting.py
+ins_pricing_gemini/modelling/tests/test_plotting_library.py
+ins_pricing_gemini/modelling/tests/test_preprocessor.py
+ins_pricing_gemini/pricing/__init__.py
+ins_pricing_gemini/pricing/calibration.py
+ins_pricing_gemini/pricing/data_quality.py
+ins_pricing_gemini/pricing/exposure.py
+ins_pricing_gemini/pricing/factors.py
+ins_pricing_gemini/pricing/monitoring.py
+ins_pricing_gemini/pricing/rate_table.py
+ins_pricing_gemini/production/__init__.py
+ins_pricing_gemini/production/drift.py
+ins_pricing_gemini/production/monitoring.py
+ins_pricing_gemini/production/scoring.py
+ins_pricing_gemini/reporting/__init__.py
+ins_pricing_gemini/reporting/report_builder.py
+ins_pricing_gemini/reporting/scheduler.py
+ins_pricing_gemini/scripts/BayesOpt_incremental.py
+ins_pricing_gemini/scripts/Explain_entry.py
+ins_pricing_gemini/scripts/__init__.py
+ins_pricing_gemini/scripts/train.py
 user_packages/__init__.py
 user_packages legacy/BayesOpt.py
 user_packages legacy/BayesOpt_entry.py

ins_pricing-0.1.8/Ins_Pricing.egg-info/requires.txt

@@ -0,0 +1,42 @@
+numpy>=1.20
+pandas>=1.4
+
+[all]
+torch>=1.13
+optuna>=3.0
+xgboost>=1.6
+scikit-learn>=1.1
+statsmodels>=0.13
+joblib>=1.2
+matplotlib>=3.5
+shap>=0.41
+contextily>=1.3
+pynndescent>=0.5
+torch-geometric>=2.3
+
+[bayesopt]
+torch>=1.13
+optuna>=3.0
+xgboost>=1.6
+scikit-learn>=1.1
+statsmodels>=0.13
+joblib>=1.2
+matplotlib>=3.5
+
+[explain]
+torch>=1.13
+shap>=0.41
+scikit-learn>=1.1
+
+[geo]
+contextily>=1.3
+matplotlib>=3.5
+
+[gnn]
+torch>=1.13
+pynndescent>=0.5
+torch-geometric>=2.3
+
+[plotting]
+matplotlib>=3.5
+scikit-learn>=1.1

{ins_pricing-0.1.6 → ins_pricing-0.1.8}/Ins_Pricing.egg-info/top_level.txt

@@ -1,4 +1,4 @@
 ins_pricing
-ins_pricing v2
+ins_pricing_gemini
 user_packages
 user_packages legacy

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

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: ins_pricing
-Version: 0.1.6
+Version: 0.1.8
 Summary: Reusable modelling, pricing, governance, and reporting utilities.
 Author: meishi125478
 License: Proprietary
@@ -15,17 +15,40 @@ Requires-Python: >=3.9
 Description-Content-Type: text/markdown
 Requires-Dist: numpy>=1.20
 Requires-Dist: pandas>=1.4
-Requires-Dist: torch>=1.13
-Requires-Dist: optuna>=3.0
-Requires-Dist: xgboost>=1.6
-Requires-Dist: scikit-learn>=1.1
-Requires-Dist: statsmodels>=0.13
-Requires-Dist: joblib>=1.2
-Requires-Dist: matplotlib>=3.5
-Requires-Dist: shap>=0.41
-Requires-Dist: contextily>=1.3
-Requires-Dist: pynndescent>=0.5
-Requires-Dist: torch-geometric>=2.3
+Provides-Extra: bayesopt
+Requires-Dist: torch>=1.13; extra == "bayesopt"
+Requires-Dist: optuna>=3.0; extra == "bayesopt"
+Requires-Dist: xgboost>=1.6; extra == "bayesopt"
+Requires-Dist: scikit-learn>=1.1; extra == "bayesopt"
+Requires-Dist: statsmodels>=0.13; extra == "bayesopt"
+Requires-Dist: joblib>=1.2; extra == "bayesopt"
+Requires-Dist: matplotlib>=3.5; extra == "bayesopt"
+Provides-Extra: plotting
+Requires-Dist: matplotlib>=3.5; extra == "plotting"
+Requires-Dist: scikit-learn>=1.1; extra == "plotting"
+Provides-Extra: explain
+Requires-Dist: torch>=1.13; extra == "explain"
+Requires-Dist: shap>=0.41; extra == "explain"
+Requires-Dist: scikit-learn>=1.1; extra == "explain"
+Provides-Extra: geo
+Requires-Dist: contextily>=1.3; extra == "geo"
+Requires-Dist: matplotlib>=3.5; extra == "geo"
+Provides-Extra: gnn
+Requires-Dist: torch>=1.13; extra == "gnn"
+Requires-Dist: pynndescent>=0.5; extra == "gnn"
+Requires-Dist: torch-geometric>=2.3; extra == "gnn"
+Provides-Extra: all
+Requires-Dist: torch>=1.13; extra == "all"
+Requires-Dist: optuna>=3.0; extra == "all"
+Requires-Dist: xgboost>=1.6; extra == "all"
+Requires-Dist: scikit-learn>=1.1; extra == "all"
+Requires-Dist: statsmodels>=0.13; extra == "all"
+Requires-Dist: joblib>=1.2; extra == "all"
+Requires-Dist: matplotlib>=3.5; extra == "all"
+Requires-Dist: shap>=0.41; extra == "all"
+Requires-Dist: contextily>=1.3; extra == "all"
+Requires-Dist: pynndescent>=0.5; extra == "all"
+Requires-Dist: torch-geometric>=2.3; extra == "all"
 
 # Ins-Pricing Overview
 

{ins_pricing-0.1.6 → 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.8/ins_pricing/setup.py

@@ -0,0 +1,74 @@
+from setuptools import setup, find_packages
+
+
+def _discover_packages() -> list[str]:
+    root_packages = ["modelling", "pricing",
+                     "production", "governance", "reporting"]
+    packages = ["ins_pricing"]
+    for root in root_packages:
+        found = find_packages(where=".", include=[root, f"{root}.*"])
+        for pkg in found:
+            packages.append(f"ins_pricing.{pkg}")
+    return packages
+
+
+setup(
+    name="Ins-Pricing",
+    version="0.1.8",
+    description="Reusable modelling, pricing, governance, and reporting utilities.",
+    author="meishi125478",
+    license="Proprietary",
+    python_requires=">=3.9",
+    package_dir={"ins_pricing": "."},
+    packages=_discover_packages(),
+    install_requires=[
+        "numpy>=1.20",
+        "pandas>=1.4",
+    ],
+    extras_require={
+        "bayesopt": [
+            "torch>=1.13",
+            "optuna>=3.0",
+            "xgboost>=1.6",
+            "scikit-learn>=1.1",
+            "statsmodels>=0.13",
+            "joblib>=1.2",
+            "matplotlib>=3.5",
+        ],
+        "plotting": [
+            "matplotlib>=3.5",
+            "scikit-learn>=1.1",
+        ],
+        "explain": [
+            "torch>=1.13",
+            "shap>=0.41",
+            "scikit-learn>=1.1",
+        ],
+        "geo": [
+            "contextily>=1.3",
+            "matplotlib>=3.5",
+        ],
+        "gnn": [
+            "torch>=1.13",
+            "pynndescent>=0.5",
+            "torch-geometric>=2.3",
+        ],
+        "all": [
+            "torch>=1.13",
+            "optuna>=3.0",
+            "xgboost>=1.6",
+            "scikit-learn>=1.1",
+            "statsmodels>=0.13",
+            "joblib>=1.2",
+            "matplotlib>=3.5",
+            "shap>=0.41",
+            "contextily>=1.3",
+            "pynndescent>=0.5",
+            "torch-geometric>=2.3",
+        ],
+    },
+    include_package_data=True,
+    package_data={"ins_pricing": ["**/*.json", "**/*.md"]},
+    exclude_package_data={"ins_pricing": [
+        "modelling/demo/*", "modelling/demo/**/*"]},
+)

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

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "ins_pricing"
-version = "0.1.6"
+version = "0.1.8"
 description = "Reusable modelling, pricing, governance, and reporting utilities."
 readme = "README.md"
 requires-python = ">=3.9"
@@ -22,6 +22,37 @@ classifiers = [
 dependencies = [
   "numpy>=1.20",
   "pandas>=1.4",
+]
+
+[project.optional-dependencies]
+bayesopt = [
+  "torch>=1.13",
+  "optuna>=3.0",
+  "xgboost>=1.6",
+  "scikit-learn>=1.1",
+  "statsmodels>=0.13",
+  "joblib>=1.2",
+  "matplotlib>=3.5",
+]
+plotting = [
+  "matplotlib>=3.5",
+  "scikit-learn>=1.1",
+]
+explain = [
+  "torch>=1.13",
+  "shap>=0.41",
+  "scikit-learn>=1.1",
+]
+geo = [
+  "contextily>=1.3",
+  "matplotlib>=3.5",
+]
+gnn = [
+  "torch>=1.13",
+  "pynndescent>=0.5",
+  "torch-geometric>=2.3",
+]
+all = [
   "torch>=1.13",
   "optuna>=3.0",
   "xgboost>=1.6",

ins_pricing-0.1.6/Ins_Pricing.egg-info/requires.txt

@@ -1,13 +0,0 @@
-numpy>=1.20
-pandas>=1.4
-torch>=1.13
-optuna>=3.0
-xgboost>=1.6
-scikit-learn>=1.1
-statsmodels>=0.13
-joblib>=1.2
-matplotlib>=3.5
-shap>=0.41
-contextily>=1.3
-pynndescent>=0.5
-torch-geometric>=2.3

ins_pricing-0.1.6/ins_pricing/setup.py

@@ -1,41 +0,0 @@
-from setuptools import setup, find_packages
-
-
-def _discover_packages() -> list[str]:
-    root_packages = ["modelling", "pricing", "production", "governance", "reporting"]
-    packages = ["ins_pricing"]
-    for root in root_packages:
-        found = find_packages(where=".", include=[root, f"{root}.*"])
-        for pkg in found:
-            packages.append(f"ins_pricing.{pkg}")
-    return packages
-
-
-setup(
-    name="ins_pricing",
-    version="0.1.6",
-    description="Reusable modelling, pricing, governance, and reporting utilities.",
-    author="meishi125478",
-    license="Proprietary",
-    python_requires=">=3.9",
-    package_dir={"ins_pricing": "."},
-    packages=_discover_packages(),
-    install_requires=[
-        "numpy>=1.20",
-        "pandas>=1.4",
-        "torch>=1.13",
-        "optuna>=3.0",
-        "xgboost>=1.6",
-        "scikit-learn>=1.1",
-        "statsmodels>=0.13",
-        "joblib>=1.2",
-        "matplotlib>=3.5",
-        "shap>=0.41",
-        "contextily>=1.3",
-        "pynndescent>=0.5",
-        "torch-geometric>=2.3",
-    ],
-    include_package_data=True,
-    package_data={"ins_pricing": ["**/*.json", "**/*.md"]},
-    exclude_package_data={"ins_pricing": ["modelling/demo/*", "modelling/demo/**/*"]},
-)