ignore overload, as previously

Author: hauntsaninja

mypy/plugins/functools.py

@@ -13,6 +13,7 @@
     AnyType,
     CallableType,
     Instance,
+    Overloaded,
     Type,
     TypeOfAny,
     UnboundType,
@@ -124,6 +125,9 @@ def partial_new_callback(ctx: mypy.plugin.FunctionContext) -> Type:
     if len(ctx.arg_types[0]) != 1:
         return ctx.default_return_type
 
+    if isinstance(ctx.arg_types[0][0], Overloaded):
+        # TODO: handle overloads, just fall back to whatever the non-plugin code does
+        return ctx.default_return_type
     fn_type = ctx.api.extract_callable_type(ctx.arg_types[0][0], ctx=ctx.default_return_type)
     if fn_type is None:
         return ctx.default_return_type

test-data/unit/check-functools.test

@@ -261,6 +261,23 @@ def main2(f: CallbackProto) -> None:
     p("a")  # E: Argument 1 to "__call__" of "CallbackProto" has incompatible type "str"; expected "int"
 [builtins fixtures/dict.pyi]
 
+[case testFunctoolsPartialOverload]
+from typing import overload
+import functools
+
+@overload
+def foo(a: int, b: str) -> int: ...
+@overload
+def foo(a: str, b: int) -> str: ...
+def foo(*a, **k): ...
+
+p1 = functools.partial(foo)
+reveal_type(p1(1, "a"))  # N: Revealed type is "builtins.int"
+reveal_type(p1("a", 1))  # N: Revealed type is "builtins.int"
+p1(1, 2)  # TODO: false negative
+p1("a", "b")  # TODO: false negative
+[builtins fixtures/dict.pyi]
+
 [case testFunctoolsPartialTypeGuard]
 import functools
 from typing_extensions import TypeGuard