Fix support for multiple hooks in attach_hook

`Axon.attach_hook` should accumulate hooks instead of overwriting the `node.hooks` variable with the most recent hook
elixir-nx · May 24, 2023 · cb0b347 · cb0b347
1 parent bcafdef
commit cb0b347
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/lib/axon.ex b/lib/axon.ex
@@ -3139,7 +3139,7 @@ defmodule Axon do
 
     updated_nodes =
       Map.update!(nodes, id, fn axon_node ->
-        %{axon_node | hooks: [{on_event, mode, fun}]}
+        %{axon_node | hooks: [{on_event, mode, fun} | axon_node.hooks]}
       end)
 
     %{axon | nodes: updated_nodes}