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

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.4",
     description="Reusable modelling, pricing, governance, and reporting utilities.",
     author="meishi125478",
     license="Proprietary",

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

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

{ins_pricing-0.4.2.dist-info → ins_pricing-0.4.4.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=skNhJD6j6gCSzDBOl7RI2YikhLMySbTS1qQ2MHRzJg0,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=o8uyDvEsJ3NrVy-VRo7c3DoKcJQsFSK4vY29vYMkrSk,37093
 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.4.dist-info/METADATA,sha256=OIsdS7Kh7LadvJ2EqK_ZVxNqEAtWEU2X3IqvEi2eD4I,6263
+ins_pricing-0.4.4.dist-info/WHEEL,sha256=wUyA8OaulRlbfwMtmQsvNngGrxQHAvkKcvRmdizlJi0,92
+ins_pricing-0.4.4.dist-info/top_level.txt,sha256=haZuNQpHKNBEPZx3NjLnHp8pV3I_J9QG8-HyJn00FA0,12
+ins_pricing-0.4.4.dist-info/RECORD,,