Fix support for multiple hooks in attach_hook (#495)

elixir-nx · May 31, 2023 · 006650c · 006650c
1 parent bcafdef
commit 006650c
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}