diff --git a/silverback/application.py b/silverback/application.py index b04eea34..11bc4978 100644 --- a/silverback/application.py +++ b/silverback/application.py @@ -139,7 +139,7 @@ def on_startup(self) -> Callable: def do_something_on_startup(startup_state): ... # Reprocess missed events or blocks """ - return self.broker_task_decorator(TaskType.STARTUP) + return self.broker_task_decorator(TaskType.STARTUP) # type: ignore[arg-type] def on_shutdown(self) -> Callable: """ @@ -151,7 +151,7 @@ def on_shutdown(self) -> Callable: def do_something_on_shutdown(): ... # Record final state of app """ - return self.broker_task_decorator(TaskType.SHUTDOWN) + return self.broker_task_decorator(TaskType.SHUTDOWN) # type: ignore[arg-type] def on_worker_startup(self) -> Callable: """ @@ -210,7 +210,10 @@ def on_( else: self.poll_settings["_blocks_"] = {"start_block": start_block} - return self.broker_task_decorator(TaskType.NEW_BLOCKS, container=container) + return self.broker_task_decorator( + TaskType.NEW_BLOCKS, # type: ignore[arg-type] + container=container, + ) elif isinstance(container, ContractEvent) and isinstance( container.contract, ContractInstance @@ -229,7 +232,10 @@ def on_( else: self.poll_settings[key] = {"start_block": start_block} - return self.broker_task_decorator(TaskType.EVENT_LOG, container=container) + return self.broker_task_decorator( + TaskType.EVENT_LOG, # type: ignore[arg-type] + container=container, + ) # TODO: Support account transaction polling # TODO: Support mempool polling diff --git a/silverback/types.py b/silverback/types.py index c6c19b86..183e7460 100644 --- a/silverback/types.py +++ b/silverback/types.py @@ -4,10 +4,11 @@ from typing_extensions import Self # Introduced 3.11 try: - from enum import StrEnum # Only Python 3.11+ + # NOTE: Only Python 3.11+ + from enum import StrEnum # type: ignore[attr-defined] except ImportError: - from backports.strenum import StrEnum # type: ignore[no-redef] + from backports.strenum import StrEnum # type: ignore[no-redef,import-not-found] class TaskType(StrEnum):