gitea_admin/sientia-dataops-laborious_temporal
1
0
Fork 0
Code Issues Pull Requests Actions 1 Packages Projects Releases Wiki Activity

SIENTIAPDE-1231

Browse Source 
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
Download Patch File Download Diff File Expand all files Collapse all files

6
laborious/utils/repository/model_repository.py
View File

@@ -455,12 +455,12 @@ class MLFlowRepository():
self.logger.custom_info(f"Index type: {index_type}", metadata) 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 # Check if all in index are of the same type
if not all(isinstance(i, index_type) for i in index): if not all(isinstance(i, index_type) for i in index):
raise ValueError( raise ValueError(
f"{message}") f"{message}. Elements are {','.join(map(type, index))}")
# Check type and converts to DATETIME_FORMAT_WITH_TZ # Check type and converts to DATETIME_FORMAT_WITH_TZ
if index_type == str: if index_type == str:
@@ -896,7 +896,7 @@ class MLFlowRepository():
data_path = f"{model_temp_path}/retrain_data.csv" 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( self.logger.custom_info(
f"Starting model upload for {experiment_name} with run name {current_run_name}", metadata) f"Starting model upload for {experiment_name} with run name {current_run_name}", metadata)

6
values.yaml
View File

@@ -3,7 +3,7 @@
# Declare variables to be passed into your templates. # Declare variables to be passed into your templates.
# This will set the replicaset count more information can be found here: https://kubernetes.io/docs/concepts/workloads/controllers/replicaset/ # This will set the replicaset count more information can be found here: https://kubernetes.io/docs/concepts/workloads/controllers/replicaset/
replicaCount: 1 replicaCount: 2
# This sets the container image more information can be found here: https://kubernetes.io/docs/concepts/containers/images/ # This sets the container image more information can be found here: https://kubernetes.io/docs/concepts/containers/images/
image: image:
@@ -11,7 +11,7 @@ image:
# This sets the pull policy for images. # This sets the pull policy for images.
pullPolicy: Always pullPolicy: Always
# Overrides the image tag whose default is the chart appVersion. # Overrides the image tag whose default is the chart appVersion.
tag: "0.0.2" tag: "0.0.3"
0# This is for the secrets for pulling an image from a private repository more information can be found here: https://kubernetes.io/docs/tasks/configure-pod-container/pull-image-private-registry/ 0# This is for the secrets for pulling an image from a private repository more information can be found here: https://kubernetes.io/docs/tasks/configure-pod-container/pull-image-private-registry/
imagePullSecrets: imagePullSecrets:
@@ -214,7 +214,7 @@ env:
value: "1" value: "1"
- name: MINIO_ENDPOINT_URL - name: MINIO_ENDPOINT_URL
value: "http://sientia-minio-minio.sientia.svc.cluster.local:9000" value: "http://minio.minio.svc.cluster.local:9000"
- name: MINIO_ACCESS_KEY - name: MINIO_ACCESS_KEY
value: "admin" value: "admin"
- name: MINIO_SECRET_KEY - name: MINIO_SECRET_KEY