Fix bug when registering hooks with execution context

tapis-project · Feb 10, 2024 · bcd53f6 · bcd53f6
1 parent 0ae4cb1
commit bcd53f6
Showing 1 changed file with 1 addition and 3 deletions.
diff --git a/src/engine/src/owe_python_sdk/runtime/execution_context.py b/src/engine/src/owe_python_sdk/runtime/execution_context.py
@@ -24,9 +24,7 @@ def add_hook(self, exit_code: int, hook: callable, *args, **kwargs):
         if not callable(hook):
             raise TypeError("Error registering hook. Argument 'hook' must be callable")
 
-        self._exit_hooks(exit_code, partial(hook, *args, **kwargs))
-
-        return self
+        self._exit_hooks[exit_code].append(partial(hook, *args, **kwargs))
 
 
     def get_input(self, input_id, default=None):