From a4e48c0f845e776c240793d6d7e76a33be29059e Mon Sep 17 00:00:00 2001
From: Samuel Moors <samuel.moors@vub.be>
Date: Wed, 9 Oct 2024 17:25:47 +0200
Subject: [PATCH] update lammps hook name

---
 eessi/testsuite/tests/apps/lammps/lammps.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/eessi/testsuite/tests/apps/lammps/lammps.py b/eessi/testsuite/tests/apps/lammps/lammps.py
index 22e70d3e..f8b41bd4 100644
--- 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':