Function wrapper

tensorrt_llm/_ray_utils.py:39–41

(self, *args, **kwargs)

    @functools.wraps(func)
    def wrapper(self, *args, **kwargs):
        with self.engine.control_action():
            return func(self, *args, **kwargs)

    return wrapper