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:
@@ -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)
|
||||
|
||||
Reference in New Issue
Block a user