SIENTIAPDE-1222

Update image tag in values.yaml and enhance debug logging in Gates and MLFlow activities

- Updated the image tag in values.yaml from '0.0.1' to '0.0.2'.
- Improved debug logging in the Gates activity to format input data and filters for better readability.
- Enhanced MLFlow activity logging to include formatted output for raw and transformed response data, ensuring consistent logging format.

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:

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)
 

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

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.