U DAf@sFddlZddlZddlZddlmZeeZde e ddddZ dS)N)loggingF)seeddeterministic_cudnnreturnc Cst|tj|t|tjd<z:ddl}|||j ||r\d|j j _ d|j j _ Wn4ttfk r}ztjd|dW5d}~XYnXdS)a Setting multiple seeds to make runs reproducible. Important: Enabling `deterministic_cudnn` gives you full reproducibility with CUDA, but might slow down your training (see https://pytorch.org/docs/stable/notes/randomness.html#cudnn) ! :param seed:number to use as seed :param deterministic_cudnn: Enable for full reproducibility when using CUDA. Caution: might slow down training. ZPYTHONHASHSEEDrNTFzQCould not set PyTorch seed because torch is not installed. Exception: {exception}) exception)randomrnpstrosenvirontorchZ manual_seedZcudaZmanual_seed_allbackendsZcudnnZ deterministicZ benchmark ImportErrorModuleNotFoundErrorloggerinfo)rrr excr?/tmp/pip-unpacked-wheel-z752163x/haystack/testing/test_utils.py set_all_seedss     r)F) r rZnumpyrZhaystackr getLogger__name__rintboolrrrrrs