fix lint

Author: kossman

src/corva/__init__.py

@@ -1,22 +1,4 @@
-from .api import Api
-from .handlers import scheduled, stream, task, partial_rerun_merge
-from .logger import CORVA_LOGGER as Logger
-from .models.rerun import RerunDepth, RerunDepthRange, RerunTime, RerunTimeRange
-from .models.scheduled.scheduled import (
-    ScheduledDataTimeEvent,
-    ScheduledDepthEvent,
-    ScheduledNaturalTimeEvent,
-)
-from .models.stream.stream import (
-    StreamDepthEvent,
-    StreamDepthRecord,
-    StreamTimeEvent,
-    StreamTimeRecord,
-)
-from .models.merge.merge import PartialRerunMergeEvent
-from .models.task import TaskEvent
-from .service.cache_sdk import UserRedisSdk as Cache
-from .shared import SECRETS as secrets
+from .models.scheduled.scheduled import ScheduledDataTimeEvent
 
 
 def __getattr__(name):

src/corva/validate_app_init.py

@@ -2,7 +2,7 @@
 import functools
 import json
 import os
-from typing import Any, Optional, Type, Dict
+from typing import Any, Dict, Optional, Type
 
 import pydantic
 
@@ -34,15 +34,17 @@ def guess_event_type(aws_event: Any) -> Optional[Type[RawBaseEvent]]:
 def validate_manifested_type(manifest: Dict[str, Any], app_decorator_type: str) -> None:
     manifest_app_type = manifest.get("application", {}).get("type")
     if manifest_app_type and manifest_app_type != app_decorator_type:
-        raise RuntimeError(f'Application with type "{manifest_app_type}" '
-                           f'can\'t invoke a "{app_decorator_type}" handler')
+        raise RuntimeError(
+            f'Application with type "{manifest_app_type}" '
+            f'can\'t invoke a "{app_decorator_type}" handler'
+        )
 
 
 def validate_event_payload(aws_event, app_decorator_type) -> None:
     base_event_cls_to_app_type_mapping: Dict[str, Type[RawBaseEvent]] = {
         "task": RawTaskEvent,
         "stream": RawStreamEvent,
-        "scheduled": RawScheduledEvent
+        "scheduled": RawScheduledEvent,
     }
 
     if event_cls := guess_event_type(aws_event):
@@ -51,7 +53,9 @@ def validate_event_payload(aws_event, app_decorator_type) -> None:
             #   run mode for existing app types
             return
 
-        if not issubclass(event_cls, base_event_cls_to_app_type_mapping.get(app_decorator_type)):
+        if not issubclass(
+            event_cls, base_event_cls_to_app_type_mapping.get(app_decorator_type)
+        ):
             raise RuntimeError(
                 f'Application with type "{app_decorator_type}" '
                 f'was invoked with "{event_cls}" event type'

tests/unit/test_apps_event_types.py

@@ -99,14 +99,11 @@ def lambda_handler(_event, _api):
 
     particular_app = app_decorator(lambda_handler)
 
-    mocked_manifest = {
-        "application":
-            {
-                "type": manifested_app_type
-            }
-    }
-
-    with mock.patch("corva.validate_app_init.read_manifest", return_value=mocked_manifest):
+    mocked_manifest = {"application": {"type": manifested_app_type}}
+
+    with mock.patch(
+        "corva.validate_app_init.read_manifest", return_value=mocked_manifest
+    ):
         with pytest.raises(RuntimeError):
             particular_app(
                 {},

tests/unit/test_partial_rerun_merge_app.py

@@ -3,7 +3,6 @@
 from uuid import uuid4
 
 import pytest
-from pydantic import ValidationError
 
 import corva
 
@@ -52,7 +51,10 @@ def partial_rerun_merge_app(event, api, asset_cache, rerun_asset_cache):
 
     with pytest.raises(RuntimeError) as e:
         stream_app(raw_event, context)
-    assert 'Application with type "stream" was invoked with "unknown" event type' in str(e.value)
+    assert (
+        'Application with type "stream" was invoked with "unknown" event type'
+        in str(e.value)
+    )
 
 
 def test_merge_event_handler_called_from_stream_app_returns_expected_cache_values(