ins-pricing 0.4.5__py3-none-any.whl → 0.5.1__py3-none-any.whl

ins_pricing/utils/__init__.py

@@ -1,83 +1,117 @@
-"""Shared utilities for the ins_pricing package.
-
-This module provides common utilities used across all submodules:
-- Logging: Unified logging system with configurable levels
-- Metrics: PSI calculation, model evaluation metrics
-- Paths: Path resolution and data loading utilities
-- Device: GPU/CPU device management for PyTorch models
-
-Example:
-    >>> from ins_pricing.utils import get_logger, psi_report
-    >>> logger = get_logger("my_module")
-    >>> logger.info("Processing started")
-"""
-
-from __future__ import annotations
-
-# =============================================================================
+"""Shared utilities for the ins_pricing package.
+
+This module provides common utilities used across all submodules:
+- Logging: Unified logging system with configurable levels
+- Metrics: PSI calculation, model evaluation metrics
+- Paths: Path resolution and data loading utilities
+- IO: File helpers and parameter loading
+- Numerics: EPS, tweedie loss, adaptive batch sizing
+- Device: GPU/CPU device management for PyTorch models
+
+Example:
+    >>> from ins_pricing.utils import get_logger, psi_report
+    >>> logger = get_logger("my_module")
+    >>> logger.info("Processing started")
+"""
+
+from __future__ import annotations
+
+# =============================================================================
 # Logging utilities
 # =============================================================================
-from .logging import get_logger, configure_logging
-
-# =============================================================================
-# Metric utilities (PSI, model evaluation)
-# =============================================================================
-from .metrics import (
-    psi_numeric,
-    psi_categorical,
-    population_stability_index,
-    psi_report,
-    MetricFactory,
-)
-
-# =============================================================================
-# Path utilities
-# =============================================================================
-from .paths import (
-    resolve_path,
-    resolve_dir_path,
-    resolve_data_path,
-    load_dataset,
-    coerce_dataset_types,
-    dedupe_preserve_order,
-    build_model_names,
-    parse_model_pairs,
-    fingerprint_file,
-    PLOT_MODEL_LABELS,
-    PYTORCH_TRAINERS,
-)
-
-# =============================================================================
-# Device management (GPU/CPU)
-# =============================================================================
-from .device import (
-    DeviceManager,
-    GPUMemoryManager,
-)
-
-__all__ = [
+from ins_pricing.utils.logging import get_logger, configure_logging, log_print
+
+# =============================================================================
+# Metric utilities (PSI, model evaluation)
+# =============================================================================
+from ins_pricing.utils.metrics import (
+    psi_numeric,
+    psi_categorical,
+    population_stability_index,
+    psi_report,
+    MetricFactory,
+)
+
+# =============================================================================
+# Numerical helpers
+# =============================================================================
+from ins_pricing.utils.numerics import (
+    EPS,
+    set_global_seed,
+    compute_batch_size,
+    tweedie_loss,
+)
+
+# =============================================================================
+# Feature inference
+# =============================================================================
+from ins_pricing.utils.features import infer_factor_and_cate_list
+
+# =============================================================================
+# IO helpers
+# =============================================================================
+from ins_pricing.utils.io import IOUtils, csv_to_dict, ensure_parent_dir
+
+# =============================================================================
+# Path utilities
+# =============================================================================
+from ins_pricing.utils.paths import (
+    resolve_path,
+    resolve_dir_path,
+    resolve_data_path,
+    load_dataset,
+    coerce_dataset_types,
+    dedupe_preserve_order,
+    build_model_names,
+    parse_model_pairs,
+    fingerprint_file,
+    PLOT_MODEL_LABELS,
+    PYTORCH_TRAINERS,
+)
+
+# =============================================================================
+# Device management (GPU/CPU)
+# =============================================================================
+from ins_pricing.utils.device import (
+    DeviceManager,
+    GPUMemoryManager,
+)
+
+__all__ = [
     # Logging
     "get_logger",
     "configure_logging",
-    # Metrics
-    "psi_numeric",
-    "psi_categorical",
-    "population_stability_index",
-    "psi_report",
-    "MetricFactory",
-    # Paths
-    "resolve_path",
-    "resolve_dir_path",
-    "resolve_data_path",
-    "load_dataset",
-    "coerce_dataset_types",
-    "dedupe_preserve_order",
-    "build_model_names",
-    "parse_model_pairs",
-    "fingerprint_file",
-    "PLOT_MODEL_LABELS",
-    "PYTORCH_TRAINERS",
-    # Device
-    "DeviceManager",
-    "GPUMemoryManager",
-]
+    "log_print",
+    # Metrics
+    "psi_numeric",
+    "psi_categorical",
+    "population_stability_index",
+    "psi_report",
+    "MetricFactory",
+    # Numerics
+    "EPS",
+    "set_global_seed",
+    "compute_batch_size",
+    "tweedie_loss",
+    # Features
+    "infer_factor_and_cate_list",
+    # IO
+    "IOUtils",
+    "csv_to_dict",
+    "ensure_parent_dir",
+    # Paths
+    "resolve_path",
+    "resolve_dir_path",
+    "resolve_data_path",
+    "load_dataset",
+    "coerce_dataset_types",
+    "dedupe_preserve_order",
+    "build_model_names",
+    "parse_model_pairs",
+    "fingerprint_file",
+    "PLOT_MODEL_LABELS",
+    "PYTORCH_TRAINERS",
+    # Device
+    "DeviceManager",
+    "GPUMemoryManager",
+]