From 15233849dbcdcaae56a27674f3b02b9efb8cca97 Mon Sep 17 00:00:00 2001
From: Khalim Conn-Kowlessar <kconnkowlessar@gmail.com>
Date: Fri, 1 Sep 2023 18:41:20 +0100
Subject: [PATCH] test adding logging

---
 .../simulation_system/handlers/predictions_app.py    | 12 +++++++++---
 1 file changed, 9 insertions(+), 3 deletions(-)

diff --git a/model_data/simulation_system/handlers/predictions_app.py b/model_data/simulation_system/handlers/predictions_app.py
index 2ecdd522..b09e2d67 100644
--- a/model_data/simulation_system/handlers/predictions_app.py
+++ b/model_data/simulation_system/handlers/predictions_app.py
@@ -1,7 +1,11 @@
 import os
-import urllib.parse
+import logging
+
 from predictions import prediction
 
+logger = logging.getLogger()
+logger.setLevel(logging.INFO)
+
 RUNTIME_ENVIRONMENT = os.environ.get("RUNTIME_ENVIRONMENT", "dev")
 
 
@@ -10,6 +14,8 @@ def handler(event, context):
     Take in event and trigger the prediction pipeline
     """
 
+    logger.info("received event: " + str(event))
+
     # Assuming a file in a bucket landing for now?
     # Assuming we have a model to use
 
@@ -30,8 +36,8 @@ def handler(event, context):
         return storage_filepath
 
     except (Exception, KeyError, ValueError) as e:
-        print("Prediction failed")
-        print(e)
+        logger.info("Prediction failed")
+        logger.info(e)
 
 
 if __name__ == "__main__":