diff --git a/laborious/activities/gates.py b/laborious/activities/gates.py index cbebebe..c88b59a 100644 --- a/laborious/activities/gates.py +++ b/laborious/activities/gates.py @@ -206,8 +206,8 @@ class Gates(BaseActivity): filter_output = [] - self.debug(create_sample_dict( - data, max_items=5, max_depth=2), metadata) + self.debug(f"Input data: \n {create_sample_dict( + data, max_items=5, max_depth=2)}", metadata) self.debug(f"Filters: {filters}", metadata) comments = [] @@ -293,7 +293,7 @@ class Gates(BaseActivity): filter_output = [] self.debug(f"Input data:\n {data.head(5).to_string()}", metadata) - self.debug(create_sample_dict(filters), metadata) + self.debug(f"Filters: \n {create_sample_dict(filters)}", metadata) for fil, config in filters.items(): if fil not in mlflow_content_filter_functions: diff --git a/laborious/activities/mlflow.py b/laborious/activities/mlflow.py index 5b449e0..33d9aeb 100644 --- a/laborious/activities/mlflow.py +++ b/laborious/activities/mlflow.py @@ -161,9 +161,8 @@ class MLFlow(BaseActivity): model_name, data, model_config ) - self.debug("Transform raw response data:", metadata) - self.debug(create_sample_dict( - response_data), metadata) + self.debug(f"Transform raw response data: \n {create_sample_dict( + response_data)}", metadata) if response_data['success']: @@ -194,9 +193,8 @@ class MLFlow(BaseActivity): response_data['content'] = response_dataframe.to_dict() - self.debug("Transform response data:", metadata) - self.debug(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) @@ -248,9 +246,8 @@ class MLFlow(BaseActivity): model_name, data, model_config ) - self.debug("Prediction response data:", metadata) - self.debug(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) diff --git a/laborious/utils/repository/model_repository.py b/laborious/utils/repository/model_repository.py index 07f0fc5..a4bd28f 100644 --- a/laborious/utils/repository/model_repository.py +++ b/laborious/utils/repository/model_repository.py @@ -27,8 +27,16 @@ ARTIFACTS_PATH = "./tmp/artifacts" class MLFlowRepository(): def __init__(self, host: str, username: str, password: str, logger: Logger): + +<< << << < HEAD # set tracking uri mlflow.set_tracking_uri(host) +== == == = + self.model_serving = ModelServing(tracking_uri=host, + username=username, password=password, + logger=logger) + self.logger = logger +>>>>>> > eefd081(SIENTIAPDE-1222) environ["MLFLOW_TRACKING_USERNAME"] = username environ["MLFLOW_TRACKING_PASSWORD"] = password diff --git a/values.yaml b/values.yaml index b2ae84e..4eb83d7 100644 --- a/values.yaml +++ b/values.yaml @@ -11,9 +11,9 @@ image: # This sets the pull policy for images. pullPolicy: Always # Overrides the image tag whose default is the chart appVersion. - tag: "0.0.1" + tag: "0.0.2" -# This is for the secrets for pulling an image from a private repository more information can be found here: https://kubernetes.io/docs/tasks/configure-pod-container/pull-image-private-registry/ +0# This is for the secrets for pulling an image from a private repository more information can be found here: https://kubernetes.io/docs/tasks/configure-pod-container/pull-image-private-registry/ imagePullSecrets: - name: docker-hub-secret # This is to override the chart name.