False positive for Union[type, ...] and inspect.isclass()

Bug Report

When a variable's type is declared as a union that includes type (or typing.Type[...]), mypy does not take into account whether usage of that variable occurs inside of an if inspect.isclass(...) conditional block, leading to false positives.

To Reproduce

gist
playground

from inspect import isclass

registry: dict[str, type] = {}

def decorator_factory(arg: str | type):
    if isclass(arg):
        registry[arg.__name__.lower()] = arg
        return arg
    else:
        def decorator(cls: type):
            registry[arg] = cls
            return cls
        return decorator
        
@decorator_factory
class Foo:
    pass
        
@decorator_factory("alt")
class Bar:
    pass

Expected Behavior**

mypy reports no errors

Actual Behavior

main.py:11: error: Invalid index type "str | type" for "dict[str, type]"; expected type "str"  [index]
Found 1 error in 1 file (checked 1 source file)

Workaround

Using typing.cast to narrow the union explicitly.

gist
playground

from inspect import isclass
from typing import cast, TYPE_CHECKING

registry: dict[str, type] = {}

def decorator_factory(arg: str | type):
    if isclass(arg):
        registry[arg.__name__.lower()] = arg
        return arg
    else:
        if TYPE_CHECKING:
            arg = cast(str, arg)
        
        def decorator(cls: type):
            registry[arg] = cls
            return cls
        return decorator
        
@decorator_factory
class Foo:
    pass
        
@decorator_factory("alt")
class Bar:
    pass

Your Environment

• Mypy version used: 1.7.1
• Mypy command-line flags: (mypy-play defaults)
• Mypy configuration options from mypy.ini (and other config files): (mypy-play defaults)
• Python version used: 3.12

Führe zunächst den verlinkten Reproducer mit mypy aus und vergleiche den gemeldeten Fehler mit dem erwarteten Verhalten rund um inspect.isclass() und Union[type, ...]. Verfolge anschließend den Pfad der Typverengung für inspect.isclass und füge eine Regressionstestabdeckung für den Reproducer hinzu. Überprüfe, dass mypy ohne die cast-Umgehungslösung keine Fehler meldet.