SIENTIAPDE-1579: Switched from replace logic to validation + mapping

This commit is contained in:
Kou Kinoshita
2026-02-18 18:57:41 -03:00
parent e7e851dbb5
commit 14f3fbadbe
5 changed files with 105 additions and 34 deletions

View File

@@ -1,6 +1,8 @@
from dataclasses import dataclass
from typing import Any
from model_manager.sientia.models import validate_frontend_date_format
# Model name constants
MODEL_LINEAR_REGRESSION = 'Linear Regression'
MODEL_POLYNOMIAL_REGRESSION = 'Polynomial Regression'
@@ -217,6 +219,7 @@ class TrainModelParams:
self._validate_intervals_and_dates()
self._validate_limits()
self._validate_required_strings()
self._validate_date_format()
def _validate_numeric_ranges(self) -> None:
"""Validate numeric parameters are within acceptable ranges."""
@@ -327,3 +330,8 @@ class TrainModelParams:
if not self.experiment_name.strip():
raise ValueError('experiment_name cannot be empty or whitespace')
def _validate_date_format(self) -> None:
"""Validate date_format is one of the allowed frontend formats when set."""
if self.date_format:
validate_frontend_date_format(self.date_format)