o 2wi@s2ddlZddlZddlZddlZejddZdS)Nccs|jd}|durd}n|jd}t}tj}t}t |tj |t ||Vt |tj |t |dS)a\ Pytest fixture that ensures deterministic RNG behavior. After the test finishes, it restores the previous RNG state. Example usage:: >>> def my_test(deterministic_rng): ... x = torch.randn(10, 5) # always has the same values You can also set random seed like this:: >>> @pytest.mark.seed(1337) ... def my_test(deterministic_rng): ... x = torch.randn(10, 5) .. note: Learn more about pytest fixtures setup/teardown here: https://docs.pytest.org/en/latest/how-to/fixtures.html#teardown-cleanup-aka-fixture-finalization seedNr)nodeget_closest_markerargstorch get_rng_statenprandom get_stategetstate manual_seedrsetstate set_state set_rng_state)requestmarkerSEED torch_statenp_statepy_staterR/home/ubuntu/sommelier/.venv/lib/python3.10/site-packages/lhotse/testing/random.pydeterministic_rngs        r)r numpyrpytestrfixturerrrrrs