more mypyc

xor2003 · Apr 23, 2024 · 91f59cc · 91f59cc
1 parent 8878da1
commit 91f59cc
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/masm2c/Token.py b/masm2c/Token.py
@@ -64,7 +64,7 @@ def find_tokens(expr: Any, lookfor: str) -> Union[list[str], list[lark.tree.Tree
                 if result := Token.find_tokens(expr[i], lookfor):
                     l += result
         if not l:
-            l = None
+            return None
         return l
 
 
@@ -106,8 +106,8 @@ def __init__(self, *args, **kwargs) -> None:
         self.element_number = 1
         self.element_size_guess = 0
         self.ptr_size = 0
-        self.mods = set()
-        self.registers = set()
+        self.mods: set[str] = set()
+        self.registers: set[str] = set()
         self.segment_register = "ds"
         self.segment_overriden = False
         self.original_type = ""