:jRdZddlmZddlmZddlmZmZed dededfdZ y ) z1Utilities to help with reproducibility of models.) Generator)contextmanager_collect_rng_states_set_rng_states include_cudareturn)NNNc#@Kt|}dt|yw)aA context manager that resets the global random state on exit to what it was before entering. It supports isolating the states for PyTorch, Numpy, and Python built-in random number generators. Args: include_cuda: Whether to allow this function to also control the `torch.cuda` random number generator. Set this to ``False`` when using the function in a forked process where CUDA re-initialization is prohibited. Example: >>> import torch >>> torch.manual_seed(1) # doctest: +ELLIPSIS >>> with isolate_rng(): ... [torch.rand(1) for _ in range(3)] [tensor([0.7576]), tensor([0.2793]), tensor([0.4031])] >>> torch.rand(1) tensor([0.7576]) Nr)rstatess u/home/mcse/projects/srt_converter/srt-converter-venv/lib/python3.12/site-packages/lightning/pytorch/utilities/seed.py isolate_rngr s,! .F FsN)T) __doc__collections.abcr contextlibrlightning.fabric.utilities.seedrrboolr r rs;8%%Pdi8H.Ir