diff --git a/python/tvm/runtime/disco/session.py b/python/tvm/runtime/disco/session.py index 8d0ff57a32bf..b166bd82e9e5 100644 --- a/python/tvm/runtime/disco/session.py +++ b/python/tvm/runtime/disco/session.py @@ -364,8 +364,8 @@ def __init__(self, num_workers: int, entrypoint: str) -> None: self.__init_handle_by_constructor__( _ffi_api.SessionProcess, # type: ignore # pylint: disable=no-member num_workers, - entrypoint, "runtime.disco.create_process_pool", + entrypoint, )