addressing CR comments

Author: corva-alex-s

src/corva/handlers.py

@@ -571,6 +571,7 @@ def _merge_events(aws_event: Any, data_transformation_type: Type[RawBaseEvent])
     Merge happens differently, depending on app type.
     """
     if data_transformation_type == RawScheduledEvent:
+        # scheduled event
         if not isinstance(aws_event[0], dict):
             aws_event = list(itertools.chain(*aws_event))
         is_depth = aws_event[0]["scheduler_type"] == SchedulerType.data_depth_milestone

tests/unit/test_stream_app.py

@@ -634,7 +634,7 @@ def stream_app(event, api, cache):
     event = []
     timestamp = 1
     # generate 3 events with 2 records each
-    for _ in range(1, 4):
+    for _ in range(3):
         event.extend(
             [
                 RawStreamTimeEvent(