Merge efbdd65 into aabb2da

mrocklin · Sep 15, 2017 · 4556088 · 4556088
2 parents aabb2da + efbdd65
commit 4556088
Showing 1 changed file with 11 additions and 1 deletion.
diff --git a/multipledispatch/core.py b/multipledispatch/core.py
@@ -54,11 +54,21 @@ def dispatch(*types, **kwargs):
     on_ambiguity = kwargs.get('on_ambiguity', ambiguity_warn)
 
     types = tuple(types)
+
     def _(func):
+        nonlocal types
+        if len(types) == 0:
+            sig = inspect.signature(func)
+            annotations = tuple(
+                param.annotation for param in sig.parameters.values())
+            if all(ann is not inspect.Parameter.empty for ann in annotations):
+                types = annotations
+
         name = func.__name__
 
         if ismethod(func):
-            dispatcher = inspect.currentframe().f_back.f_locals.get(name,
+            dispatcher = inspect.currentframe().f_back.f_locals.get(
+                name,
                 MethodDispatcher(name))
         else:
             if name not in namespace: