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

SIENTIAPDE-1199

Browse Source 
Enhance logging in MLFlow activity to include formatted JSON response data

- Added JSON import for better formatting of response data in debug logs.
- Updated logging statements to output transformed and prediction response data as pretty-printed JSON, improving readability and debugging context.
This commit is contained in:
vitor-aignosi
2025-08-19 12:12:15 -03:00
parent 8012ff778c
commit 46e21c7538
1 changed files with 4 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
laborious/activities/mlflow.py
View File

@@ -1,3 +1,4 @@
import json
from temporalio import activity, workflow from temporalio import activity, workflow
@@ -66,8 +67,8 @@ class MLFlow(BaseActivity):
response_data = self.model_monitoring_repository.transform( response_data = self.model_monitoring_repository.transform(
model_name, data, model_retention) model_name, data, model_retention)
self.debug("Response data:", metadata) self.debug("Transform response data:", metadata)
self.debug(response_data, metadata) self.debug(json.dumps(response_data, indent=4), metadata)
return response_data return response_data
@@ -97,7 +98,7 @@ class MLFlow(BaseActivity):
model_name, data, model_retention) model_name, data, model_retention)
self.debug("Prediction response data:", metadata) self.debug("Prediction response data:", metadata)
self.debug(response_data, metadata) self.debug(json.dumps(response_data, indent=4), metadata)
return response_data return response_data