ins-pricing 0.4.2__tar.gz → 0.4.4__tar.gz

{ins_pricing-0.4.2 → ins_pricing-0.4.4}/PKG-INFO

@@ -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 → ins_pricing-0.4.4}/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-0.4.2 → ins_pricing-0.4.4}/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: