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

SIENTIAPDE-1222

Browse Source 
Refactor MLFlow debug logging for improved readability

- Reformatted debug logging statements in the MLFlow activity to enhance clarity and consistency.
- Ensured that the output of raw and transformed response data is presented in a more readable format, maintaining the use of create_sample_dict for better visualization.
This commit is contained in:
vitor-aignosi
2025-09-16 10:29:35 -03:00
parent 0bdbef00b7
commit 8095e37710
1 changed files with 6 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

12
laborious/activities/mlflow.py
View File

@@ -161,8 +161,8 @@ class MLFlow(BaseActivity):
model_name, data, model_config model_name, data, model_config
) )
self.debug(f"Transform raw response data: \n {create_sample_dict( self.debug(
response_data)}", metadata) f"Transform raw response data: \n {create_sample_dict(response_data)}", metadata)
if response_data['success']: if response_data['success']:
@@ -193,8 +193,8 @@ class MLFlow(BaseActivity):
response_data['content'] = response_dataframe.to_dict() response_data['content'] = response_dataframe.to_dict()
self.debug(f"Transform response data: \n {create_sample_dict( self.debug(
response_data)}", metadata) f"Transform response data: \n {create_sample_dict(response_data)}", metadata)
self.info("Data transformed successfully", metadata) self.info("Data transformed successfully", metadata)
@@ -246,8 +246,8 @@ class MLFlow(BaseActivity):
model_name, data, model_config model_name, data, model_config
) )
self.debug(f"Prediction response data: \n {create_sample_dict( self.debug(
response_data)}", metadata) f"Prediction response data: \n {create_sample_dict(response_data)}", metadata)
self.info("Data predicted successfully", metadata) self.info("Data predicted successfully", metadata)