async def METHOD_NAME(self, partitions, fn, all_nodes=False):
    fn_with_args = functools.partial(
        self._wrapped.METHOD_NAME, partitions, fn, all_nodes
    )
    return await sync_to_async(fn_with_args, self._pool)