diff --git a/laborious/activities/mlflow.py b/laborious/activities/mlflow.py index 33d9aeb..e0e7cb3 100644 --- a/laborious/activities/mlflow.py +++ b/laborious/activities/mlflow.py @@ -161,8 +161,8 @@ class MLFlow(BaseActivity): model_name, data, model_config ) - self.debug(f"Transform raw response data: \n {create_sample_dict( - response_data)}", metadata) + self.debug( + f"Transform raw response data: \n {create_sample_dict(response_data)}", metadata) if response_data['success']: @@ -193,8 +193,8 @@ class MLFlow(BaseActivity): response_data['content'] = response_dataframe.to_dict() - self.debug(f"Transform response data: \n {create_sample_dict( - response_data)}", metadata) + self.debug( + f"Transform response data: \n {create_sample_dict(response_data)}", metadata) self.info("Data transformed successfully", metadata) @@ -246,8 +246,8 @@ class MLFlow(BaseActivity): model_name, data, model_config ) - self.debug(f"Prediction response data: \n {create_sample_dict( - response_data)}", metadata) + self.debug( + f"Prediction response data: \n {create_sample_dict(response_data)}", metadata) self.info("Data predicted successfully", metadata)