From 73b7e846009a47436562fe5c5cbb230bf287f71c Mon Sep 17 00:00:00 2001 From: Caspar van Leeuwen Date: Tue, 7 May 2024 11:24:30 +0200 Subject: [PATCH] Can't combine generators with plus, so use chain --- eessi/testsuite/tests/apps/PyTorch/PyTorch_torchvision.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/eessi/testsuite/tests/apps/PyTorch/PyTorch_torchvision.py b/eessi/testsuite/tests/apps/PyTorch/PyTorch_torchvision.py index 68223cef..575527cf 100644 --- 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'