Add typing in so mypy can actually catch the incorrect BASIC_ALLOWED_…

…ATTRS
danthedeckie · Oct 31, 2024 · c4e1a61 · c4e1a61
1 parent 32f1bfc
commit c4e1a61
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/simpleeval.py b/simpleeval.py
@@ -107,6 +107,7 @@
 import sys
 import warnings
 from random import random
+from typing import Type, Dict, Set, Union
 
 ########################################
 # Module wide 'globals'
@@ -144,7 +145,7 @@
 
 # Opt-in type safety experiment. Will be opt-out in 2.x
 
-BASIC_ALLOWED_ATTRS = {
+BASIC_ALLOWED_ATTRS: Dict[Union[Type, None], Set] = {
     int: {
         "as_integer_ratio",
         "bit_length",