mypy - typ zwracanej wartości przez funkcję taki sam jak typ argumentu funkcji

Question

Hej, jak mogę powiedzieć mypy że typ zwracany powinien być taki sam jak ten przekazany w parametrze _type do funkcji read_number bez bawienia się w Any?

Chodzi mi o takie coś

def read_number(prompt: str = 'Enter the number (cannot be zero): ', *,
                _type: type = float) -> _type (?) wywala błąd undefined name _type:
    while True:
        try:
            return _type(input(prompt))
        except ValueError:
            print('Error: not a number. Try again...')

Funkcja ma zwracać taki sam typ jak ten zdefiniowany w zmiennej _type.

Comments

Polecamy użyć TypeVar z bound=type i adnotacji Callable

przed funkcją
T = TypeVar("T")
w definicji funkcji
prompt: str = "Enter the number (cannot be zero): ", *, _type: Callable[[str], T] = float, ) -> T