addressing CR comments

corva-ai · Nov 30, 2023 · e1a854a · e1a854a
1 parent 44d14ff
commit e1a854a
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 1 deletion.
diff --git a/src/corva/handlers.py b/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

diff --git a/tests/unit/test_stream_app.py b/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(