python/mypy

Wrong return type inferred for overload with a type variable

eltoder opened this issue · 1 comments

Bug Report

from typing import TypeVar, overload

_T = TypeVar("_T")

@overload
def getenv(key: str) -> str | None: ...

@overload
def getenv(key: str, default: _T) -> str | _T: ...

def getenv(key, default):
    return key


def test(custom_var: str) -> str:
    return getenv(custom_var) or getenv("V", "D")

https://mypy-play.net/?mypy=latest&python=3.12&gist=4d959cd7069a8495533bfeca1cd40729

Actual Behavior

main.py:16: error: Incompatible return value type (got "str | None", expected "str")  [return-value]

Expected Behavior

This should type check without errors, since we have an or with a call that always returns str.

It works if I replace type variable with just str:

@overload
def getenv(key: str, default: str) -> str: ...

but something about _T and the union upsets it.

Your Environment

  • Mypy version used: 1.13.0
  • Mypy command-line flags: none
  • Mypy configuration options from mypy.ini (and other config files): none
  • Python version used: 3.10-3.13

A slightly simpler example without @overload:

from typing import TypeVar

_T = TypeVar("_T")

def foo(key: str) -> str | None: return key

def bar(key: str, default: _T) -> str | _T: return key


def test(custom_var: str) -> str:
    return foo(custom_var) or bar("V", "D")

https://mypy-play.net/?mypy=latest&python=3.12&gist=0afee5cdbf09752bb5f6a5ee9f3e35f4