ins-pricing 0.3.1__tar.gz → 0.3.3__tar.gz

{ins_pricing-0.3.1 → ins_pricing-0.3.3}/PKG-INFO

@@ -1,162 +1,162 @@
-Metadata-Version: 2.4
-Name: ins_pricing
-Version: 0.3.1
-Summary: Reusable modelling, pricing, governance, and reporting utilities.
-Author: meishi125478
-License: Proprietary
-Keywords: pricing,insurance,bayesopt,ml
-Classifier: Programming Language :: Python :: 3
-Classifier: Programming Language :: Python :: 3 :: Only
-Classifier: Programming Language :: Python :: 3.9
-Classifier: License :: Other/Proprietary License
-Classifier: Operating System :: OS Independent
-Classifier: Intended Audience :: Developers
-Requires-Python: >=3.9
-Description-Content-Type: text/markdown
-Requires-Dist: numpy>=1.20
-Requires-Dist: pandas>=1.4
-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"
-
-# Insurance-Pricing
-
-A reusable toolkit for insurance modeling, pricing, governance, and reporting.
-
-## Overview
-
-Insurance-Pricing (ins_pricing) is an enterprise-grade Python library designed for machine learning model training, pricing calculations, and model governance workflows in the insurance industry.
-
-### Core Modules
-
-| Module | Description |
-|--------|-------------|
-| **modelling** | ML model training (GLM, XGBoost, ResNet, FT-Transformer, GNN) and model interpretability (SHAP, permutation importance) |
-| **pricing** | Factor table construction, numeric binning, premium calibration, exposure calculation, PSI monitoring |
-| **production** | Model prediction, batch scoring, data drift detection, production metrics monitoring |
-| **governance** | Model registry, version management, approval workflows, audit logging |
-| **reporting** | Report generation (Markdown format), report scheduling |
-| **utils** | Data validation, performance profiling, device management, logging configuration |
-
-### Quick Start
-
-```python
-# Model training with Bayesian optimization
-from ins_pricing import bayesopt as ropt
-
-model = ropt.BayesOptModel(
-    train_data, test_data,
-    model_name='my_model',
-    resp_nme='target',
-    weight_nme='weight',
-    factor_nmes=feature_list,
-    cate_list=categorical_features,
-)
-model.bayesopt_xgb(max_evals=100)      # Train XGBoost
-model.bayesopt_resnet(max_evals=50)    # Train ResNet
-model.bayesopt_ft(max_evals=50)        # Train FT-Transformer
-
-# Pricing: build factor table
-from ins_pricing.pricing import build_factor_table
-factors = build_factor_table(
-    df,
-    factor_col='age_band',
-    loss_col='claim_amount',
-    exposure_col='exposure',
-)
-
-# Production: batch scoring
-from ins_pricing.production import batch_score
-scores = batch_score(model.trainers['xgb'].predict, df)
-
-# Model governance
-from ins_pricing.governance import ModelRegistry
-registry = ModelRegistry('models.json')
-registry.register(model_name, version, metrics=metrics)
-```
-
-### Project Structure
-
-```
-ins_pricing/
-├── cli/                    # Command-line entry points
-├── modelling/
-│   ├── core/bayesopt/     # ML model training core
-│   ├── explain/           # Model interpretability
-│   └── plotting/          # Model visualization
-├── pricing/               # Insurance pricing module
-├── production/            # Production deployment module
-├── governance/            # Model governance
-├── reporting/             # Report generation
-├── utils/                 # Utilities
-└── tests/                 # Test suite
-```
-
-### Installation
-
-```bash
-# Basic installation
-pip install ins_pricing
-
-# Full installation (all optional dependencies)
-pip install ins_pricing[all]
-
-# Install specific extras
-pip install ins_pricing[bayesopt]    # Model training
-pip install ins_pricing[explain]     # Model explanation
-pip install ins_pricing[plotting]    # Visualization
-pip install ins_pricing[gnn]         # Graph neural networks
-```
-
-#### Multi-platform & GPU installation notes
-
-- **PyTorch (CPU/GPU/MPS)**: Install the correct PyTorch build for your platform/GPU first (CUDA on
-  Linux/Windows, ROCm on supported AMD platforms, or MPS on Apple Silicon). Then install the
-  optional extras you need (e.g., `bayesopt`, `explain`, or `gnn`). This avoids pip pulling a
-  mismatched wheel.
-- **Torch Geometric (GNN)**: `torch-geometric` often requires platform-specific wheels (e.g.,
-  `torch-scatter`, `torch-sparse`). Follow the official PyG installation instructions for your
-  CUDA/ROCm/CPU environment, then install `ins_pricing[gnn]`.
-- **Multi-GPU**: Training code will use CUDA when available and can enable multi-GPU via
-  `torch.distributed`/`DataParallel` where supported. On Windows, CUDA DDP is not supported and will
-  fall back to single-GPU or DataParallel where possible.
-
-### Requirements
-
-- Python >= 3.9
-- Core dependencies: numpy >= 1.20, pandas >= 1.4
-
-### License
-
-Proprietary
+Metadata-Version: 2.4
+Name: ins_pricing
+Version: 0.3.3
+Summary: Reusable modelling, pricing, governance, and reporting utilities.
+Author: meishi125478
+License: Proprietary
+Keywords: pricing,insurance,bayesopt,ml
+Classifier: Programming Language :: Python :: 3
+Classifier: Programming Language :: Python :: 3 :: Only
+Classifier: Programming Language :: Python :: 3.9
+Classifier: License :: Other/Proprietary License
+Classifier: Operating System :: OS Independent
+Classifier: Intended Audience :: Developers
+Requires-Python: >=3.9
+Description-Content-Type: text/markdown
+Requires-Dist: numpy>=1.20
+Requires-Dist: pandas>=1.4
+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"
+
+# Insurance-Pricing
+
+A reusable toolkit for insurance modeling, pricing, governance, and reporting.
+
+## Overview
+
+Insurance-Pricing (ins_pricing) is an enterprise-grade Python library designed for machine learning model training, pricing calculations, and model governance workflows in the insurance industry.
+
+### Core Modules
+
+| Module | Description |
+|--------|-------------|
+| **modelling** | ML model training (GLM, XGBoost, ResNet, FT-Transformer, GNN) and model interpretability (SHAP, permutation importance) |
+| **pricing** | Factor table construction, numeric binning, premium calibration, exposure calculation, PSI monitoring |
+| **production** | Model prediction, batch scoring, data drift detection, production metrics monitoring |
+| **governance** | Model registry, version management, approval workflows, audit logging |
+| **reporting** | Report generation (Markdown format), report scheduling |
+| **utils** | Data validation, performance profiling, device management, logging configuration |
+
+### Quick Start
+
+```python
+# Model training with Bayesian optimization
+from ins_pricing import bayesopt as ropt
+
+model = ropt.BayesOptModel(
+    train_data, test_data,
+    model_name='my_model',
+    resp_nme='target',
+    weight_nme='weight',
+    factor_nmes=feature_list,
+    cate_list=categorical_features,
+)
+model.bayesopt_xgb(max_evals=100)      # Train XGBoost
+model.bayesopt_resnet(max_evals=50)    # Train ResNet
+model.bayesopt_ft(max_evals=50)        # Train FT-Transformer
+
+# Pricing: build factor table
+from ins_pricing.pricing import build_factor_table
+factors = build_factor_table(
+    df,
+    factor_col='age_band',
+    loss_col='claim_amount',
+    exposure_col='exposure',
+)
+
+# Production: batch scoring
+from ins_pricing.production import batch_score
+scores = batch_score(model.trainers['xgb'].predict, df)
+
+# Model governance
+from ins_pricing.governance import ModelRegistry
+registry = ModelRegistry('models.json')
+registry.register(model_name, version, metrics=metrics)
+```
+
+### Project Structure
+
+```
+ins_pricing/
+├── cli/                    # Command-line entry points
+├── modelling/
+│   ├── core/bayesopt/     # ML model training core
+│   ├── explain/           # Model interpretability
+│   └── plotting/          # Model visualization
+├── pricing/               # Insurance pricing module
+├── production/            # Production deployment module
+├── governance/            # Model governance
+├── reporting/             # Report generation
+├── utils/                 # Utilities
+└── tests/                 # Test suite
+```
+
+### Installation
+
+```bash
+# Basic installation
+pip install ins_pricing
+
+# Full installation (all optional dependencies)
+pip install ins_pricing[all]
+
+# Install specific extras
+pip install ins_pricing[bayesopt]    # Model training
+pip install ins_pricing[explain]     # Model explanation
+pip install ins_pricing[plotting]    # Visualization
+pip install ins_pricing[gnn]         # Graph neural networks
+```
+
+#### Multi-platform & GPU installation notes
+
+- **PyTorch (CPU/GPU/MPS)**: Install the correct PyTorch build for your platform/GPU first (CUDA on
+  Linux/Windows, ROCm on supported AMD platforms, or MPS on Apple Silicon). Then install the
+  optional extras you need (e.g., `bayesopt`, `explain`, or `gnn`). This avoids pip pulling a
+  mismatched wheel.
+- **Torch Geometric (GNN)**: `torch-geometric` often requires platform-specific wheels (e.g.,
+  `torch-scatter`, `torch-sparse`). Follow the official PyG installation instructions for your
+  CUDA/ROCm/CPU environment, then install `ins_pricing[gnn]`.
+- **Multi-GPU**: Training code will use CUDA when available and can enable multi-GPU via
+  `torch.distributed`/`DataParallel` where supported. On Windows, CUDA DDP is not supported and will
+  fall back to single-GPU or DataParallel where possible.
+
+### Requirements
+
+- Python >= 3.9
+- Core dependencies: numpy >= 1.20, pandas >= 1.4
+
+### License
+
+Proprietary

ins_pricing-0.3.3/ins_pricing/cli/BayesOpt_entry.py

@@ -0,0 +1,56 @@
+"""Thin wrapper for the BayesOpt CLI entry point.
+
+The main implementation lives in bayesopt_entry_runner.py.
+"""
+
+from __future__ import annotations
+
+from pathlib import Path
+import json
+import os
+import sys
+
+if __package__ in {None, ""}:
+    repo_root = Path(__file__).resolve().parents[2]
+    if str(repo_root) not in sys.path:
+        sys.path.insert(0, str(repo_root))
+
+def _apply_env_from_config(argv: list[str]) -> None:
+    if "--config-json" not in argv:
+        return
+    idx = argv.index("--config-json")
+    if idx + 1 >= len(argv):
+        return
+    raw_path = argv[idx + 1]
+    try:
+        cfg_path = Path(raw_path).expanduser()
+        if not cfg_path.is_absolute():
+            cfg_path = cfg_path.resolve()
+        if not cfg_path.exists():
+            script_dir = Path(__file__).resolve().parents[1]
+            candidate = (script_dir / raw_path).resolve()
+            if candidate.exists():
+                cfg_path = candidate
+        if not cfg_path.exists():
+            return
+        cfg = json.loads(cfg_path.read_text(encoding="utf-8", errors="replace"))
+        env = cfg.get("env", {})
+        if isinstance(env, dict):
+            for key, value in env.items():
+                if key is None:
+                    continue
+                os.environ.setdefault(str(key), str(value))
+    except Exception:
+        return
+
+_apply_env_from_config(sys.argv)
+
+try:
+    from .bayesopt_entry_runner import main
+except Exception:  # pragma: no cover
+    from ins_pricing.cli.bayesopt_entry_runner import main
+
+__all__ = ["main"]
+
+if __name__ == "__main__":
+    main()

{ins_pricing-0.3.1 → ins_pricing-0.3.3}/ins_pricing/cli/utils/import_resolver.py

@@ -13,6 +13,7 @@ Usage:
 from __future__ import annotations
 
 import importlib
+import os
 import sys
 from dataclasses import dataclass, field
 from pathlib import Path
@@ -70,14 +71,39 @@ class ResolvedImports:
     plot_loss_curve: Optional[Callable] = None
 
 
+def _debug_imports_enabled() -> bool:
+    value = os.environ.get("BAYESOPT_DEBUG_IMPORTS")
+    if value is None:
+        return False
+    return str(value).strip().lower() in {"1", "true", "yes", "y", "on"}
+
+
 def _try_import(module_path: str, attr_name: Optional[str] = None) -> Optional[Any]:
     """Attempt to import a module or attribute, returning None on failure."""
     try:
         module = importlib.import_module(module_path)
         if attr_name:
-            return getattr(module, attr_name, None)
-        return module
-    except Exception:
+            result = getattr(module, attr_name, None)
+        else:
+            result = module
+        if _debug_imports_enabled():
+            origin = getattr(module, "__file__", None)
+            origin = origin or getattr(module, "__path__", None)
+            print(
+                f"[BAYESOPT_DEBUG_IMPORTS] imported {module_path}"
+                f"{'::' + attr_name if attr_name else ''} from {origin}",
+                file=sys.stderr,
+                flush=True,
+            )
+        return result
+    except Exception as exc:
+        if _debug_imports_enabled():
+            print(
+                f"[BAYESOPT_DEBUG_IMPORTS] failed import {module_path}"
+                f"{'::' + attr_name if attr_name else ''}: {exc.__class__.__name__}: {exc}",
+                file=sys.stderr,
+                flush=True,
+            )
         return None
 
 

{ins_pricing-0.3.1 → ins_pricing-0.3.3}/ins_pricing/cli/utils/notebook_utils.py

@@ -8,9 +8,9 @@ from pathlib import Path
 from typing import Iterable, List, Optional, Sequence, cast
 
 try:
-    from .cli_config import add_config_json_arg  # type: ignore
+    from .cli_config import add_config_json_arg, set_env  # type: ignore
 except Exception:  # pragma: no cover
-    from cli_config import add_config_json_arg  # type: ignore
+    from cli_config import add_config_json_arg, set_env  # type: ignore
 
 
 def _find_ins_pricing_dir(cwd: Optional[Path] = None) -> Path:
@@ -261,6 +261,7 @@ def run_from_config(config_json: str | Path) -> subprocess.CompletedProcess:
     if not config_path.is_absolute():
         config_path = (pkg_dir / config_path).resolve() if (pkg_dir / config_path).exists() else config_path.resolve()
     raw = json.loads(config_path.read_text(encoding="utf-8", errors="replace"))
+    set_env(raw.get("env", {}))
     runner = cast(dict, raw.get("runner") or {})
 
     mode = str(runner.get("mode") or "entry").strip().lower()

{ins_pricing-0.3.1 → ins_pricing-0.3.3}/ins_pricing/modelling/core/bayesopt/__init__.py

@@ -4,6 +4,10 @@ from __future__ import annotations
 
 import torch
 
+from ins_pricing.utils.torch_compat import disable_torch_dynamo_if_requested
+
+disable_torch_dynamo_if_requested()
+
 from .config_preprocess import (
     BayesOptConfig,
     DatasetPreprocessor,

{ins_pricing-0.3.1 → ins_pricing-0.3.3}/ins_pricing/modelling/core/bayesopt/models/model_ft_components.py

@@ -106,31 +106,58 @@ class ScaledTransformerEncoderLayer(nn.Module):
         self.res_scale_attn = residual_scale_attn
         self.res_scale_ffn = residual_scale_ffn
 
-    def forward(self, src, src_mask=None, src_key_padding_mask=None):
+    def forward(self, src, src_mask=None, src_key_padding_mask=None, is_causal: Optional[bool] = None, **_kwargs):
         # Input tensor shape: (batch, seq_len, d_model).
         x = src
 
         if self.norm_first:
             # Pre-norm before attention.
-            x = x + self._sa_block(self.norm1(x), src_mask,
-                                   src_key_padding_mask)
+            x = x + self._sa_block(
+                self.norm1(x),
+                src_mask,
+                src_key_padding_mask,
+                is_causal=is_causal,
+            )
             x = x + self._ff_block(self.norm2(x))
         else:
             # Post-norm (usually disabled).
             x = self.norm1(
-                x + self._sa_block(x, src_mask, src_key_padding_mask))
+                x + self._sa_block(
+                    x,
+                    src_mask,
+                    src_key_padding_mask,
+                    is_causal=is_causal,
+                )
+            )
             x = self.norm2(x + self._ff_block(x))
 
         return x
 
-    def _sa_block(self, x, attn_mask, key_padding_mask):
+    def _sa_block(self, x, attn_mask, key_padding_mask, *, is_causal: Optional[bool] = None):
         # Self-attention with residual scaling.
-        attn_out, _ = self.self_attn(
-            x, x, x,
-            attn_mask=attn_mask,
-            key_padding_mask=key_padding_mask,
-            need_weights=False
-        )
+        if is_causal is None:
+            attn_out, _ = self.self_attn(
+                x, x, x,
+                attn_mask=attn_mask,
+                key_padding_mask=key_padding_mask,
+                need_weights=False,
+            )
+        else:
+            try:
+                attn_out, _ = self.self_attn(
+                    x, x, x,
+                    attn_mask=attn_mask,
+                    key_padding_mask=key_padding_mask,
+                    need_weights=False,
+                    is_causal=is_causal,
+                )
+            except TypeError:
+                attn_out, _ = self.self_attn(
+                    x, x, x,
+                    attn_mask=attn_mask,
+                    key_padding_mask=key_padding_mask,
+                    need_weights=False,
+                )
         return self.res_scale_attn * self.dropout1(attn_out)
 
     def _ff_block(self, x):
@@ -313,4 +340,3 @@ class MaskedTabularDataset(Dataset):
             None if self.X_cat_true is None else self.X_cat_true[idx],
             None if self.cat_mask is None else self.cat_mask[idx],
         )
-