wait_for() + partial

I noticed that wait_for does not count with passing a partial without a message. partial() hides the function in its .func parameter.

This prevents raising an error.