SIENTIAPDE-994

Refactor logging in MLFlow and Gates activities; remove print statement in FormatAndExportPrediction; update data handling in PredictionProcess; delete unused redis-feeder script.

laborious/utils/repository/model_repository.py

@@ -13,7 +13,6 @@ import traceback
 import mlflow
 import pandas as pd
 from sientia.ModelServing import ModelServing
-from pathlib import Path
 
 
 class MLFlowRepository():
@@ -260,7 +259,8 @@ class MLFlowRepository():
         try:
             return {
                 'success': True,
-                'content': self.model_serving.get_cached_transform(model_name, data, model_retention).to_dict()
+                'content': self.model_serving.get_cached_transform(
+                    model_name, data, model_retention).to_dict()
             }
 
         except Exception as e:
@@ -276,7 +276,8 @@ class MLFlowRepository():
         try:
             start_time = datetime.now()
             data = self.model_serving.get_cached_predict(
-                model_name, data, model_retention)
+                model_name, data, model_retention)[-1:]
+
             end_time = datetime.now()
             data = pd.DataFrame(data, columns=['prediction'])
             data['response_time'] = (end_time - start_time).total_seconds()