update lammps hook name

xinan1911 · Oct 9, 2024 · a4e48c0 · a4e48c0
1 parent 548e9b3
commit a4e48c0
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/eessi/testsuite/tests/apps/lammps/lammps.py b/eessi/testsuite/tests/apps/lammps/lammps.py
@@ -50,8 +50,8 @@ def assert_run(self):
         return sn.assert_eq(n_atoms, 32000)
 
     @run_after('init')
-    def run_after_setup(self):
-        """hooks to run after the setup phase"""
+    def set_compute_unit(self):
+        """Set the compute unit to which tasks will be assigned """
         if self.device_type == 'cpu':
             self.compute_unit = COMPUTE_UNIT['CPU']
         elif self.device_type == 'gpu':