[ThSSKrSSKJr SSKJrJr SSKJr S/rSr \S\ S\S 4S j5r \\S\S 4S j55r g) N) Generator)contextmanager ExitStack)record worker_mainTORCH_WORKER_SERVER_SOCKET socket_pathreturn)NNNc#|# SSKJn U"U5nSv UR5 g!UR5 f=f7f)Nr) _WorkerServer)torch._C._distributed_c10dr shutdown)r r servers _/var/www/auris/envauris/lib/python3.13/site-packages/torch/distributed/elastic/control_plane.py_worker_serverrs.8 ; 'F s<'<9<c## [5n[RR[5nUbUR [ U55 Sv SSS5 g!,(df  g=f7f)aK This is a context manager that wraps your main entry function. This combines the existing ``errors.record`` logic as well as a new ``_WorkerServer`` that exposes handlers via a unix socket specified by ``Torch_WORKER_SERVER_SOCKET``. Example :: @worker_main() def main(): pass if __name__ == "__main__": main() N)rosenvirongetr enter_contextr)stackr s rrrsF, jjnn%?@  "   { ; < s A,AA A, A)%A,) rcollections.abcr contextlibrr0torch.distributed.elastic.multiprocessing.errorsr__all__rstrrrrrsv %0C : 2B(CY/0r