Can't combine generators with plus, so use chain

casparvl · May 7, 2024 · 73b7e84 · 73b7e84
1 parent 4ddfe23
commit 73b7e84
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/eessi/testsuite/tests/apps/PyTorch/PyTorch_torchvision.py b/eessi/testsuite/tests/apps/PyTorch/PyTorch_torchvision.py
@@ -1,3 +1,5 @@
+from itertools import chain
+
 import reframe as rfm
 import reframe.utility.sanity as sn
 # Added only to make the linter happy
@@ -14,7 +16,7 @@ class EESSI_PyTorch_torchvision(rfm.RunOnlyRegressionTest):
     parallel_strategy = parameter([None, 'ddp'])
     compute_device = variable(str)
     # Both torchvision and PyTorch-bundle modules have everything needed to run this test
-    module_name = parameter(find_modules('torchvision') + find_modules('PyTorch-bundle'))
+    module_name = parameter(chain(find_modules('torchvision'), find_modules('PyTorch-bundle')))
 
     descr = 'Benchmark that runs a selected torchvision model on synthetic data'