Тип Python подсказывает тип возвращаемого значения Union из кортежа типов в качестве аргумента ⇐ Python

[Anonymous]

Я пишу метод, который принимает кортеж типов и возвращает объект, который является экземпляром одного из этих типов или None. Я хотел бы ввести подсказку функции, чтобы показать, что она возвращает экземпляр одного из переданных типов или None в качестве объединения. Однако вместо этого подсказка типа указывает на возвращаемый тип как объединение наименьшего общего базового класса элементов в кортеже или None. Как это исправить?
В моем случае типы имеют общий базовый класс, но в идеале мне бы хотелось решение, которое работает с любым кортежем типов.
from typing import Union, Optional, TypeVar

class Base: ...
class A(Base): ...
class B(Base): ...
class C(Base): ...

def get_data() -> tuple[Base, ...]:
return (A(), C())

T = TypeVar("T", bound=Base)

# I have also tried typehinting the return value as Optional[Union[T]] but
# that didn't change anything.
def find_instance(types: tuple[type[T], ...]) -> Optional[T]:
for element in get_data():
if isinstance(element, types):
return element
return None

def find_instance2(types: tuple[T, ...]) -> Optional[T]:
for element in get_data():
if isinstance(element, types):
return element
return None

# The final typehints I'm looking for
a: Optional[Base] = find_instance((Base,)) # Inferred as Optional[Base]
b: Optional[Union[B, C]] = find_instance((B, C)) # Also inferred as Optional[Base]

# Both c and d are inferred as Optional[Type[Base]]
c: Optional[Base] = find_instance2((Base,))
d: Optional[Union[B, C]] = find_instance2((B, C))

Проверка приведенного выше кода с помощью mypy дает следующие ошибки:
test.py:18: error: Incompatible return value type (got "Base", expected "Optional[T]")
test.py:23: error: Argument 2 to "isinstance" has incompatible type "Tuple[T, ...]"; expected "Union[type, Tuple[Union[type, Tuple[Any, ...]], ...]]"
test.py:24: error: Incompatible return value type (got "Base", expected "Optional[T]")
test.py:29: error: Incompatible types in assignment (expression has type "Optional[Base]", variable has type "Union[B, C, None]")
test.py:32: error: Value of type variable "T" of "find_instance2" cannot be "Type[Base]"
test.py:32: error: Incompatible types in assignment (expression has type "Optional[Type[Base]]", variable has type "Optional[Base]")
test.py:33: error: Value of type variable "T" of "find_instance2" cannot be "Type[Base]"
test.py:33: error: Incompatible types in assignment (expression has type "Optional[Type[Base]]", variable has type "Union[B, C, None]")
Found 8 errors in 1 file (checked 1 source file)


Подробнее здесь: https://stackoverflow.com/questions/684 ... s-argument