o $i@sddlZddlmZddlmZddlmZddlmZddl m Z ddl m Z edZ eeZe Gd d d Ze Gd d d ed ZdS)N nullcontext)TypeVar)BaseWorkerGroup) Singleton) DeveloperAPI)make_table_html_repr EncodedDatac@s6eZdZdZeddZeddZdefddZd S) BackendConfigz4Parent class for configurations of training backend.cCtSN)BackendselfrN/home/ubuntu/veenaModal/venv/lib/python3.10/site-packages/ray/train/backend.py backend_clszBackendConfig.backend_clscCr r rrrrrtrain_func_contextrz BackendConfig.train_func_contextreturncCst|t|jdS)N)objtitle)rtype__name__rrrr _repr_html_szBackendConfig._repr_html_N) r __module__ __qualname____doc__propertyrrstrrrrrrr s  r c@sTeZdZUdZdZeed<dedefddZ dedefdd Z dedefd d Z d S) r aSingleton for distributed communication backend. Attributes: share_cuda_visible_devices: If True, each worker process will have CUDA_VISIBLE_DEVICES set as the visible device IDs of all workers on the same node for this training instance. If False, each worker will have CUDA_VISIBLE_DEVICES set to the device IDs allocated by Ray for that worker. Fshare_cuda_visible_devices worker_groupbackend_configcCdS)z Logic for starting this backend.Nrrr!r"rrron_start-rzBackend.on_startcCr#)z$Logic for shutting down the backend.Nrr$rrr on_shutdown1rzBackend.on_shutdowncCr#)z\Logic ran right before training is started. Session API is available at this point.Nrr$rrron_training_start5szBackend.on_training_startN) rrrrr bool__annotations__rr r%r&r'rrrrr s  r ) metaclass)logging contextlibrtypingr%ray.train._internal.base_worker_grouprray.train._internal.utilsrray.util.annotationsr ray.widgetsrr getLoggerrloggerr r rrrrs