Python Pass Function As Argument Type Hint