diff --git a/laborious/activities/mlflow.py b/laborious/activities/mlflow.py index e0e7cb3..aae09ee 100644 --- a/laborious/activities/mlflow.py +++ b/laborious/activities/mlflow.py @@ -162,7 +162,7 @@ class MLFlow(BaseActivity): ) 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']: @@ -194,7 +194,7 @@ class MLFlow(BaseActivity): response_data['content'] = response_dataframe.to_dict() 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) @@ -247,7 +247,7 @@ class MLFlow(BaseActivity): ) 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)