From 1af42d7d1e3f02d78063e02b5a603e3dd59837eb Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Adrian=20W=C3=A4lchli?= Date: Sat, 8 May 2021 20:03:51 +0200 Subject: [PATCH] fix 1.9 test (#7441) --- tests/trainer/test_dataloaders.py | 6 +++++- 1 file changed, 5 insertions(+), 1 deletion(-) diff --git a/tests/trainer/test_dataloaders.py b/tests/trainer/test_dataloaders.py index d988943c06088..6f78f125754b5 100644 --- a/tests/trainer/test_dataloaders.py +++ b/tests/trainer/test_dataloaders.py @@ -803,8 +803,12 @@ def _user_worker_init_fn(_): pass +@RunIf(max_torch="1.8.9") def test_missing_worker_init_fn(): - """ Test that naive worker seed initialization leads to undesired random state in subprocesses. """ + """ + Test that naive worker seed initialization leads to undesired random state in subprocesses. + PyTorch 1.9+ does not have this issue. + """ dataset = NumpyRandomDataset() seed_everything(0)