ins-pricing 0.4.2__py3-none-any.whl → 0.4.3__py3-none-any.whl

ins_pricing/modelling/core/bayesopt/config_preprocess.py

@@ -195,6 +195,7 @@ class BayesOptConfig:
     cache_predictions: bool = False
     prediction_cache_dir: Optional[str] = None
     prediction_cache_format: str = "parquet"
+    dataloader_workers: Optional[int] = None
 
     def __post_init__(self) -> None:
         """Validate configuration after initialization."""
@@ -210,6 +211,12 @@ class BayesOptConfig:
             errors.append(
                 f"task_type must be one of {valid_task_types}, got '{self.task_type}'"
             )
+        if self.dataloader_workers is not None:
+            try:
+                if int(self.dataloader_workers) < 0:
+                    errors.append("dataloader_workers must be >= 0 when provided.")
+            except (TypeError, ValueError):
+                errors.append("dataloader_workers must be an integer when provided.")
         # Validate loss_name
         try:
             normalized_loss = normalize_loss_name(self.loss_name, self.task_type)

ins_pricing/modelling/core/bayesopt/trainers/trainer_base.py

@@ -306,6 +306,19 @@ class TrainerBase:
         self.enable_distributed_optuna: bool = False
         self._distributed_forced_params: Optional[Dict[str, Any]] = None
 
+    def _apply_dataloader_overrides(self, model: Any) -> Any:
+        """Apply dataloader-related overrides from config to a model."""
+        cfg = getattr(self.ctx, "config", None)
+        if cfg is None:
+            return model
+        workers = getattr(cfg, "dataloader_workers", None)
+        if workers is not None:
+            model.dataloader_workers = int(workers)
+        profile = getattr(cfg, "resource_profile", None)
+        if profile:
+            model.resource_profile = str(profile)
+        return model
+
     def _export_preprocess_artifacts(self) -> Dict[str, Any]:
         dummy_columns: List[str] = []
         if getattr(self.ctx, "train_oht_data", None) is not None:

ins_pricing/modelling/core/bayesopt/trainers/trainer_ft.py

@@ -163,6 +163,7 @@ class FTTrainer(TrainerBase):
             num_numeric_tokens=num_numeric_tokens,
             loss_name=loss_name,
         )
+        model = self._apply_dataloader_overrides(model)
         model.set_params(model_params)
         try:
             return float(model.fit_unsupervised(
@@ -248,7 +249,7 @@ class FTTrainer(TrainerBase):
                 requested_heads=params.get("n_heads")
             )
 
-            return FTTransformerSklearn(
+            model = FTTransformerSklearn(
                 model_nme=self.ctx.model_nme,
                 num_cols=self.ctx.num_features,
                 cat_cols=self.ctx.cate_list,
@@ -266,7 +267,10 @@ class FTTrainer(TrainerBase):
                 use_ddp=self.ctx.config.use_ft_ddp,
                 num_numeric_tokens=num_numeric_tokens,
                 loss_name=loss_name,
-            ).set_params({"_geo_params": geo_params_local} if geo_enabled else {})
+            )
+            model = self._apply_dataloader_overrides(model)
+            model.set_params({"_geo_params": geo_params_local} if geo_enabled else {})
+            return model
 
         def fit_predict(model, X_train, y_train, w_train, X_val, y_val, w_val, trial_obj):
             geo_train = geo_val = None
@@ -353,6 +357,7 @@ class FTTrainer(TrainerBase):
                 weight_decay=float(resolved_params.get("weight_decay", 0.0)),
                 loss_name=loss_name,
             )
+            tmp_model = self._apply_dataloader_overrides(tmp_model)
             tmp_model.set_params(resolved_params)
             geo_train_full = self.ctx.train_geo_tokens
             geo_train = None if geo_train_full is None else geo_train_full.iloc[train_idx]
@@ -387,6 +392,7 @@ class FTTrainer(TrainerBase):
             weight_decay=float(resolved_params.get("weight_decay", 0.0)),
             loss_name=loss_name,
         )
+        self.model = self._apply_dataloader_overrides(self.model)
         if refit_epochs is not None:
             self.model.epochs = int(refit_epochs)
         self.model.set_params(resolved_params)
@@ -460,6 +466,7 @@ class FTTrainer(TrainerBase):
                 weight_decay=float(resolved_params.get("weight_decay", 0.0)),
                 loss_name=loss_name,
             )
+            model = self._apply_dataloader_overrides(model)
             model.set_params(resolved_params)
 
             geo_train = geo_val = None
@@ -565,6 +572,7 @@ class FTTrainer(TrainerBase):
             num_numeric_tokens=self._resolve_numeric_tokens(),
             loss_name=loss_name,
         )
+        model = self._apply_dataloader_overrides(model)
         adaptive_heads, heads_adjusted = self._resolve_adaptive_heads(
             d_model=resolved_params.get("d_model", model.d_model),
             requested_heads=resolved_params.get("n_heads"),
@@ -728,6 +736,7 @@ class FTTrainer(TrainerBase):
             num_numeric_tokens=self._resolve_numeric_tokens(),
             loss_name=loss_name,
         )
+        self.model = self._apply_dataloader_overrides(self.model)
         resolved_params = dict(params or {})
         # Reuse supervised tuning structure params unless explicitly overridden.
         if not resolved_params and self.best_params:
@@ -797,8 +806,4 @@ class FTTrainer(TrainerBase):
             self.model,
             pred_prefix=pred_prefix,
             predict_kwargs_train=predict_kwargs_train,
-            predict_kwargs_test=predict_kwargs_test
-        )
-
-
-# =============================================================================
+          

ins_pricing/modelling/core/bayesopt/trainers/trainer_gnn.py

@@ -59,7 +59,7 @@ class GNNTrainer(TrainerBase):
                 self.ctx.config.gnn_knn_gpu_mem_overhead),
             loss_name=loss_name,
         )
-        return model
+        return self._apply_dataloader_overrides(model)
 
     def cross_val(self, trial: optuna.trial.Trial) -> float:
         base_tw_power = self.ctx.default_tweedie_power()

ins_pricing/modelling/core/bayesopt/trainers/trainer_resn.py

@@ -45,7 +45,7 @@ class ResNetTrainer(TrainerBase):
                 getattr(self.ctx.config, "resn_weight_decay", 1e-4),
             )
         )
-        return ResNetSklearn(
+        model = ResNetSklearn(
             model_nme=self.ctx.model_nme,
             input_dim=self._resolve_input_dim(),
             hidden_dim=int(params.get("hidden_dim", 64)),
@@ -64,6 +64,7 @@ class ResNetTrainer(TrainerBase):
             use_ddp=self.ctx.config.use_resn_ddp,
             loss_name=loss_name
         )
+        return self._apply_dataloader_overrides(model)
 
     # ========= Cross-validation (for BayesOpt) =========
     def cross_val(self, trial: optuna.trial.Trial) -> float:

ins_pricing/modelling/core/bayesopt/utils/torch_trainer_mixin.py

@@ -232,6 +232,14 @@ class TorchTrainerMixin:
         """Determine number of DataLoader workers."""
         if os.name == 'nt':
             return 0
+        override = getattr(self, "dataloader_workers", None)
+        if override is None:
+            override = os.environ.get("BAYESOPT_DATALOADER_WORKERS")
+        if override is not None:
+            try:
+                return max(0, int(override))
+            except (TypeError, ValueError):
+                pass
         if getattr(self, "is_ddp_enabled", False):
             return 0
         profile = profile or self._resolve_resource_profile()

ins_pricing/setup.py

@@ -3,7 +3,7 @@ from setuptools import setup, find_packages
 
 setup(
     name="ins_pricing",
-    version="0.4.2",
+    version="0.4.3",
     description="Reusable modelling, pricing, governance, and reporting utilities.",
     author="meishi125478",
     license="Proprietary",

{ins_pricing-0.4.2.dist-info → ins_pricing-0.4.3.dist-info}/METADATA

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

{ins_pricing-0.4.2.dist-info → ins_pricing-0.4.3.dist-info}/RECORD

@@ -3,7 +3,7 @@ ins_pricing/README.md,sha256=W4V2xtzM6pyQzwJPvWP7cNn-We9rxM8xrxRlBVQwoY8,3399
 ins_pricing/RELEASE_NOTES_0.2.8.md,sha256=KIJzk1jbZbZPKjwnkPSDHO_2Ipv3SP3CzCNDdf07jI0,9331
 ins_pricing/__init__.py,sha256=46j1wCdLVrgrofeBwKl-3NXTxzjbTv-w3KjW-dyKGiY,2622
 ins_pricing/exceptions.py,sha256=5fZavPV4zNJ7wPC75L215KkHXX9pRrfDAYZOdSKJMGo,4778
-ins_pricing/setup.py,sha256=z5ZF0fvEybSq_iHLYPMCVQhAYqaH1fAoXaM2fCcLDP0,1702
+ins_pricing/setup.py,sha256=Jyq-oIi6qUEbE7_hXVWWbNXvF_mVP_aNQvOjf10DSu8,1702
 ins_pricing/cli/BayesOpt_entry.py,sha256=6UBVxu36O3bXn1WC-BBi-l_W9_MqEoHmDGnwwDKNo5Q,1594
 ins_pricing/cli/BayesOpt_incremental.py,sha256=_Klr5vvNoq_TbgwrH_T3f0a6cHmA9iVJMViiji6ahJY,35927
 ins_pricing/cli/Explain_Run.py,sha256=gEPQjqHiXyXlCTKjUzwSvbAn5_h74ABgb_sEGs-YHVE,664
@@ -46,7 +46,7 @@ ins_pricing/modelling/core/bayesopt/PHASE3_REFACTORING_SUMMARY.md,sha256=B8ZEzaL
 ins_pricing/modelling/core/bayesopt/REFACTORING_SUMMARY.md,sha256=hJZKXe9-bBGJVN_5c5l8nHQ1X7NK4BbeE-uXQoH0rAM,7479
 ins_pricing/modelling/core/bayesopt/__init__.py,sha256=nj6IA0r7D5U5-hYyiwXmcp_bEtoU-hRJ_prdtRmLMg0,2070
 ins_pricing/modelling/core/bayesopt/config_components.py,sha256=OjRyM1EuSXL9_3THD1nGLRsioJs7lO_ZKVZDkUA3LX8,12156
-ins_pricing/modelling/core/bayesopt/config_preprocess.py,sha256=WaoVqxi9WcS7Xat7q91bUP65fTZJdtVfx8G558F-YvU,21270
+ins_pricing/modelling/core/bayesopt/config_preprocess.py,sha256=vjxhDuJJm-bYyfphWnsZP_O3Tgtx22WGo80myLCB4cw,21647
 ins_pricing/modelling/core/bayesopt/core.py,sha256=1m4pCrPP3iYIfU6QX3j6Eczjwz3-cD4ySzv9bll3PGg,44474
 ins_pricing/modelling/core/bayesopt/model_explain_mixin.py,sha256=jCk1zPpwgwBBCndaq-A0_cQnc4RHueh2p5cAuE9ArTo,11620
 ins_pricing/modelling/core/bayesopt/model_plotting_mixin.py,sha256=lD0rUvWV4eWatmTzMrmAUm2Flj8uAOa3R9S2JyYV94k,21807
@@ -58,11 +58,11 @@ ins_pricing/modelling/core/bayesopt/models/model_ft_trainer.py,sha256=jk9pm7IzVL
 ins_pricing/modelling/core/bayesopt/models/model_gnn.py,sha256=blCTgML-fMkHDerzwoJZPw2XnEvuwVR_U5t0YWE1lZI,32901
 ins_pricing/modelling/core/bayesopt/models/model_resn.py,sha256=Pddu0q04Sz8RwKqjP0fv4xXWd6KobwMsD47sCDBbB-Y,17581
 ins_pricing/modelling/core/bayesopt/trainers/__init__.py,sha256=ODYKjT-v4IDxu4ohGLCXY8r1-pMME9LAaNx6pmj5_38,481
-ins_pricing/modelling/core/bayesopt/trainers/trainer_base.py,sha256=3oCKA-dcGC49ZvWwD5HXf9MFV7lJKcG3McnyFSCzl88,55163
-ins_pricing/modelling/core/bayesopt/trainers/trainer_ft.py,sha256=w0cR81iVtER01NKIEXOl01qRH_qQc6p7mftZpTWrO00,35473
+ins_pricing/modelling/core/bayesopt/trainers/trainer_base.py,sha256=DOam1HLsslNQlhQt88j6LWCZekioaNzw2PbKzLwtszY,55687
+ins_pricing/modelling/core/bayesopt/trainers/trainer_ft.py,sha256=DkBHF50RK70W2LlLWKbg-NEnE2LOn1cI9d-1u8zkOD0,35816
 ins_pricing/modelling/core/bayesopt/trainers/trainer_glm.py,sha256=gMhx9IX9nz-rsf-zi9UYMtViBPD1nmQ5r8XVPGU21Ys,7912
-ins_pricing/modelling/core/bayesopt/trainers/trainer_gnn.py,sha256=nPaiKXPNsN1NT6xuMLsLsGd3AbzftgLLFU9yTj7j-_M,14225
-ins_pricing/modelling/core/bayesopt/trainers/trainer_resn.py,sha256=8O_kmdir8zJy9DNT9VASp4ZDE_fKH_-Lj63BW4CI9Z0,11799
+ins_pricing/modelling/core/bayesopt/trainers/trainer_gnn.py,sha256=vJbQTm-3ByBguZYz4gvYhXeWbUa0L7z7mxaxoamGsic,14259
+ins_pricing/modelling/core/bayesopt/trainers/trainer_resn.py,sha256=MaxCvmyybE72H38akOA_rNW3Rx9Mxb229ZRJaqIzWTA,11855
 ins_pricing/modelling/core/bayesopt/trainers/trainer_xgb.py,sha256=NlEqH0wxe5frwxXNTeymWX5_qC3_rIzF3QjDZz4RBMg,13752
 ins_pricing/modelling/core/bayesopt/utils/__init__.py,sha256=dbf4DrWOH4rABOuaZdBF7drYOBH5prjvM0TexT6DYyg,1911
 ins_pricing/modelling/core/bayesopt/utils/constants.py,sha256=0ihYxGlJ8tIElYvkhIDe5FfJShegvu29WZ_Xvfqa0iE,5790
@@ -70,7 +70,7 @@ ins_pricing/modelling/core/bayesopt/utils/distributed_utils.py,sha256=cu01dHyYE5
 ins_pricing/modelling/core/bayesopt/utils/io_utils.py,sha256=vXDlAc_taCG2joxnC6wu0jVYA76UhRbX9OT_5z_im-E,3857
 ins_pricing/modelling/core/bayesopt/utils/losses.py,sha256=yn3ggeM1NRkCzcTt_Nef_EvpD6Pb_jGs49bj-VV4uWU,3894
 ins_pricing/modelling/core/bayesopt/utils/metrics_and_devices.py,sha256=kfQZnGE8FvGfl7WsTFShGGIA_sQhp5Th9mrwUXphiNQ,21200
-ins_pricing/modelling/core/bayesopt/utils/torch_trainer_mixin.py,sha256=P6EoXkIqEgU8-6i0gXznf2ms-2o8GfMLsMLLfWPmcjI,25562
+ins_pricing/modelling/core/bayesopt/utils/torch_trainer_mixin.py,sha256=qxSAP4vQC-JTHTZ7SDPZx7ZKyJKLwcAOP0IWr4GA_l4,25881
 ins_pricing/modelling/explain/__init__.py,sha256=CPoGzGu8TTO3FOXjxoXC13VkuIDCf3YTH6L3BqJq3Ok,1171
 ins_pricing/modelling/explain/gradients.py,sha256=9TqCws_p49nFxVMcjVxe4KCZ7frezeL0uV_LCdoM5yo,11088
 ins_pricing/modelling/explain/metrics.py,sha256=K_xOY7ZrHWhbJ79RNB7eXN3VXeTe8vq68ZLH2BlZufA,5389
@@ -131,7 +131,7 @@ ins_pricing/utils/paths.py,sha256=o_tBiclFvBci4cYg9WANwKPxrMcglEdOjDP-EZgGjdQ,87
 ins_pricing/utils/profiling.py,sha256=kmbykHLcYywlZxAf_aVU8HXID3zOvUcBoO5Q58AijhA,11132
 ins_pricing/utils/torch_compat.py,sha256=UrRsqx2qboDG8WE0OmxNOi08ojwE-dCxTQh0N2s3Rgw,2441
 ins_pricing/utils/validation.py,sha256=4Tw9VUJPk0N-WO3YUqZP-xXRl1Xpubkm0vi3WzzZrv4,13348
-ins_pricing-0.4.2.dist-info/METADATA,sha256=B-DRCDLLg6vBMXvenWPsCUeA015Kmv4WSchARg3PoHk,6263
-ins_pricing-0.4.2.dist-info/WHEEL,sha256=wUyA8OaulRlbfwMtmQsvNngGrxQHAvkKcvRmdizlJi0,92
-ins_pricing-0.4.2.dist-info/top_level.txt,sha256=haZuNQpHKNBEPZx3NjLnHp8pV3I_J9QG8-HyJn00FA0,12
-ins_pricing-0.4.2.dist-info/RECORD,,
+ins_pricing-0.4.3.dist-info/METADATA,sha256=33V_YLmiYurQvj4716hMRASiIVW4MgLeDstga8N6w8g,6263
+ins_pricing-0.4.3.dist-info/WHEEL,sha256=wUyA8OaulRlbfwMtmQsvNngGrxQHAvkKcvRmdizlJi0,92
+ins_pricing-0.4.3.dist-info/top_level.txt,sha256=haZuNQpHKNBEPZx3NjLnHp8pV3I_J9QG8-HyJn00FA0,12
+ins_pricing-0.4.3.dist-info/RECORD,,