annotate resolve

michalwols · Mar 23, 2024 · 8932acd · 8932acd
1 parent ac2ac62
commit 8932acd
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/yann/config/registry.py b/yann/config/registry.py
@@ -1,4 +1,5 @@
 import typing
+from typing import Union, Tuple, Dict, Any
 from collections import defaultdict, OrderedDict
 
 from functools import partial
@@ -72,7 +73,7 @@ def __init__(self, registry: 'Registry'):
 
   def resolve(
       self,
-      x,
+      x: Union[Any, Tuple[Any, Dict]],
       required=False,
       validate=None,
       instance=True,