SIENTIAPDE-1231

Update values.yaml and model_repository.py for configuration and error handling improvements

- Increased replicaCount from 1 to 2 in values.yaml for enhanced scalability.
- Updated image tag from "0.0.2" to "0.0.3" in values.yaml to reflect the latest version.
- Enhanced error messaging in model_repository.py to include the actual index type when raising ValueError for index type validation.
This commit is contained in:
vitor-aignosi
2025-10-13 15:12:20 -03:00
parent ac9b2e0ec6
commit fdaf48ddc9
2 changed files with 6 additions and 6 deletions

View File

@@ -455,12 +455,12 @@ class MLFlowRepository():
self.logger.custom_info(f"Index type: {index_type}", metadata)
message = f"Index must be all timestamp like column. Valid formats are: pandas Timestamp, datetime, string in format {DATETIME_FORMAT_WITH_TZ}"
message = f"Index must be all timestamp like column. Valid formats are: pandas Timestamp, datetime, string in format {DATETIME_FORMAT_WITH_TZ}. Got {index_type}."
# Check if all in index are of the same type
if not all(isinstance(i, index_type) for i in index):
raise ValueError(
f"{message}")
f"{message}. Elements are {','.join(map(type, index))}")
# Check type and converts to DATETIME_FORMAT_WITH_TZ
if index_type == str:
@@ -896,7 +896,7 @@ class MLFlowRepository():
data_path = f"{model_temp_path}/retrain_data.csv"
# data.to_csv(data_path, index=True)
data.to_csv(data_path, index=True)
self.logger.custom_info(
f"Starting model upload for {experiment_name} with run name {current_run_name}", metadata)