ins-pricing 0.4.2__py3-none-any.whl → 0.4.4__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: