gigl.common.utils.connect_worker_pool

gigl.common.utils.vertex_ai_context.connect_worker_pool() → DistributedContext

Used to connect the worker pool. This function should be called by all workers to get the leader worker’s internal IP address and to ensure that the workers can all communicate with the leader worker.