diff --git a/laborious/activities/mlflow.py b/laborious/activities/mlflow.py index 0f89939..d3e918d 100644 --- a/laborious/activities/mlflow.py +++ b/laborious/activities/mlflow.py @@ -1,3 +1,4 @@ +import json from temporalio import activity, workflow @@ -66,8 +67,8 @@ class MLFlow(BaseActivity): response_data = self.model_monitoring_repository.transform( model_name, data, model_retention) - self.debug("Response data:", metadata) - self.debug(response_data, metadata) + self.debug("Transform response data:", metadata) + self.debug(json.dumps(response_data, indent=4), metadata) return response_data @@ -97,7 +98,7 @@ class MLFlow(BaseActivity): model_name, data, model_retention) self.debug("Prediction response data:", metadata) - self.debug(response_data, metadata) + self.debug(json.dumps(response_data, indent=4), metadata) return response_data