test online response proto with redshift:dynamodb

TremaMiguel · TremaMiguel · commit 7a4edbd2604e · 2022-03-12T12:09:03.000-06:00
Signed-off-by: Miguel Trejo &lt;armando.trejo.marrufo@gmail.com&gt;
diff --git a/sdk/python/feast/online_response.py b/sdk/python/feast/online_response.py
@@ -22,6 +22,18 @@
 
 TIMESTAMP_POSTFIX: str = "__ts"
 
+import logging
+import sys
+
+logger = logging.getLogger()
+logger.setLevel(logging.DEBUG)
+
+handler = logging.StreamHandler(sys.stdout)
+handler.setLevel(logging.DEBUG)
+formatter = logging.Formatter('%(asctime)s - %(name)s - %(levelname)s - %(message)s')
+handler.setFormatter(formatter)
+logger.addHandler(handler)
+
 
 class OnlineResponse:
     """
@@ -57,7 +69,12 @@ def to_dict(self, include_event_timestamps: bool = False) -> Dict[str, Any]:
 
         for result in self.proto.results:
             for idx, feature_ref in enumerate(self.proto.metadata.feature_names.val):
-                native_type_value = feast_value_type_to_python_type(result.values[idx])
+                try:
+                    native_type_value = feast_value_type_to_python_type(result.values[idx])
+                except IndexError as e:
+                    logger.info(f'ERROR -- {e}')
+                    logger.info(f'ONLINE RESPONSE PROTO: {self.proto}')
+                    raise IndexError(e)
                 if feature_ref not in response:
                     response[feature_ref] = [native_type_value]
                 else:
