SIENTIAPDE-1182

Update tests.ipynb and values.yaml for improved functionality and versioning

- Updated execution count in tests.ipynb for reproducibility.
- Modified DataFrame creation in tests.ipynb to include a timestamp column and save to CSV.
- Changed image tag in values.yaml from "0.4.4" to "0.4.5" for versioning.
- Updated GITHUB_BRANCH in values.yaml to reflect the latest branch adjustments.
This commit is contained in:
vitor-aignosi
2025-09-04 09:36:23 -03:00
parent 460bfe4073
commit da21c128f8
4 changed files with 328 additions and 47 deletions

View File

@@ -39,6 +39,7 @@ class MLFlowRepository():
"""
try:
return {
'success': True,
'content': self.model_serving.get_cached_transform(
@@ -67,12 +68,15 @@ class MLFlowRepository():
- dict: A dictionary containing the predicted data.
"""
try:
input_index = data.index
start_time = datetime.now()
data = self.model_serving.get_cached_predict(
model_name, data, model_retention)
end_time = datetime.now()
data = pd.DataFrame(data, columns=['prediction'])
data.index = input_index
data['response_time'] = (end_time - start_time).total_seconds()
return {