Skip to content

Commit b2082db

Browse files
author
Vasileios Karakasis
committed
Fix Pytorch library test
1 parent 3e01030 commit b2082db

File tree

1 file changed

+5
-3
lines changed

1 file changed

+5
-3
lines changed

hpctestlib/ml/pytorch/horovod.py

Lines changed: 5 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -65,12 +65,14 @@ class pytorch_cnn_check(rfm.RunOnlyRegressionTest, pin_prefix=True):
6565
tags = {'ml', 'cnn', 'horovod'}
6666

6767
@run_after('init')
68-
def prepare_test(self):
68+
def update_descr(self):
6969
# Get the python script
70-
script = self.executable.split()[1]
71-
7270
self.descr = (f'Distributed CNN training with PyTorch and Horovod '
7371
f'(model: {self.model})')
72+
73+
@run_before('run')
74+
def prepare_run(self):
75+
script = self.executable.split()[1]
7476
self.prerun_cmds = [
7577
f'curl -LJO https://raw.githubusercontent.com/horovod/horovod/{self.benchmark_version}/examples/pytorch/{script}', # noqa: E501
7678
]

0 commit comments

Comments
 (0)