From 3bc2b262b5cbf83842e8dcac7a41ec492704442f Mon Sep 17 00:00:00 2001 From: Stefan Allius Date: Sat, 8 Jun 2024 23:59:13 +0200 Subject: [PATCH] add more type annotations --- app/src/messages.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/app/src/messages.py b/app/src/messages.py index f54b313..dbe24b5 100644 --- a/app/src/messages.py +++ b/app/src/messages.py @@ -1,6 +1,6 @@ import logging import weakref -from typing import Callable +from typing import Callable, Generator if __name__ == "app.src.messages": @@ -45,7 +45,7 @@ def hex_dump_memory(level, info, data, num): class IterRegistry(type): - def __iter__(cls): + def __iter__(cls) -> Generator['Message', None, None]: for ref in cls._registry: obj = ref() if obj is not None: