Technically more accurate.

ThePhar · Jun 2, 2024 · 757e331 · 757e331
1 parent 5d858f2
commit 757e331
Showing 1 changed file with 5 additions and 2 deletions.
diff --git a/Utils.py b/Utils.py
@@ -458,8 +458,11 @@ class KeyedDefaultDict(collections.defaultdict):
     """defaultdict variant that uses the missing key as argument to default_factory"""
     default_factory: typing.Callable[[typing.Any], typing.Any]
 
-    def __init__(self, default_factory: typing.Callable[[Any], Any] = None, *args, **kwargs):
-        super().__init__(default_factory, *args, **kwargs)
+    def __init__(self,
+                 default_factory: typing.Callable[[Any], Any] = None,
+                 default: typing.Union[typing.Mapping, typing.Iterable, None] = None,
+                 **kwargs):
+        super().__init__(default_factory, default, **kwargs)
 
     def __missing__(self, key):
         self[key] = value = self.default_factory(key)