clear all event callbacks on close

aiidateam · Jan 28, 2021 · e0596e5 · e0596e5
1 parent 1ccd44c
commit e0596e5
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/plumpy/processes.py b/plumpy/processes.py
@@ -327,7 +327,6 @@ def _setup_event_hooks(self) -> None:
         }
         for hook, callback in event_hooks.items():
             self.add_state_event_callback(hook, callback)
-            self.add_cleanup(functools.partial(self.remove_state_event_callback, hook, callback))
 
     @property
     def creation_time(self) -> Optional[float]:
@@ -845,6 +844,7 @@ def on_close(self) -> None:
                     self.logger.exception('Exception calling cleanup method %s', cleanup)
             self._cleanups = None
         finally:
+            self._event_callbacks = {}
             self._closed = True
 
     def _fire_event(self, evt: Callable[..., Any], *args: Any, **kwargs: Any) -> None: