Autocomplete with arguments for decorated functions

Consider the following program:

def decorator(f):
    return f

def foo(x, y):
    return x + y

When I type foo, I get an autocomplete that includes the argument list x, y.

Now, suppose I decorate foo:

def decorator(f):
    return f


@decorator
def foo(x, y):
    return x + y

Now, I no longer get the argument list x, y.

Is there a way to write this code such that I still get to see the argument list? (Context: I'm a library author trying to make my library work well with PyCharm. So changes to PyCharm configuration settings don't help me.)