fix timeout for single process

pmeier · pmeier · commit c5ace3e1da04 · 2022-04-01T11:30:55.000+02:00
diff --git a/test/test_prototype_builtin_datasets.py b/test/test_prototype_builtin_datasets.py
@@ -143,7 +143,7 @@ def test_pipeline(self, test_home, dataset_mock, config, parallelism_mode):
             shuffle=True,
             num_workers=num_workers,
             parallelism_mode=parallelism_mode,
-            timeout=5,
+            timeout=5 if num_workers > 0 else 0,
         )
 
         for _ in dl:

Original file line number	Diff line number	Diff line change
`@@ -143,7 +143,7 @@ def test_pipeline(self, test_home, dataset_mock, config, parallelism_mode):`
`143`	`143`	`shuffle=True,`
`144`	`144`	`num_workers=num_workers,`
`145`	`145`	`parallelism_mode=parallelism_mode,`
`146`		`- timeout=5,`
	`146`	`+ timeout=5 if num_workers > 0 else 0,`
`147`	`147`	`)`
`148`	`148`
`149`	`149`	`for _ in dl:`