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.