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

SIENTIAPDE-1222

Browse Source 
Enhance MLFlow debug logging to limit output size

- Updated debug logging statements in the MLFlow activity to include a maximum of 5 items and a depth of 5 for the sample dictionary, improving readability and reducing log clutter for raw and transformed response data.
This commit is contained in:
vitor-aignosi
2025-09-16 10:36:43 -03:00
parent 3752170c3a
commit 39aa089384
1 changed files with 3 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
laborious/activities/mlflow.py
View File

@@ -162,7 +162,7 @@ class MLFlow(BaseActivity):
) )
self.debug( self.debug(
f"Transform raw response data: \n {create_sample_dict(response_data)}", metadata) f"Transform raw response data: \n {create_sample_dict(response_data, max_items=5, max_depth=5)}", metadata)
if response_data['success']: if response_data['success']:
@@ -194,7 +194,7 @@ class MLFlow(BaseActivity):
response_data['content'] = response_dataframe.to_dict() response_data['content'] = response_dataframe.to_dict()
self.debug( self.debug(
f"Transform response data: \n {create_sample_dict(response_data)}", metadata) f"Transform response data: \n {create_sample_dict(response_data, max_items=5, max_depth=5)}", metadata)
self.info("Data transformed successfully", metadata) self.info("Data transformed successfully", metadata)
@@ -247,7 +247,7 @@ class MLFlow(BaseActivity):
) )
self.debug( self.debug(
f"Prediction response data: \n {create_sample_dict(response_data)}", metadata) f"Prediction response data: \n {create_sample_dict(response_data, max_items=5, max_depth=5)}", metadata)
self.info("Data predicted successfully", metadata) self.info("Data predicted successfully", metadata)