python3 тип, намекающий на тип другого параметра

Question

Допустим, моя функция может принимать параметр b в виде целого числа или списка

def foo_fun(a: int, b: ??)->int:
   if isinstance(b,list):
       do something
   else 
       do something else

Как я могу использовать подсказки типов с этой конкретной настройкой?

Спасибо

Answer 1

Вы, кажется, описываете Union.

Например:

from typing import Union


def foo_fun(a: int, b: Union[int, list]) -> int:
   if isinstance(b, list):
       do something
   else 
       do something else