Fix failing formatter

elixir-nx · May 14, 2024 · b6ab577 · b6ab577
1 parent 0473cf9
commit b6ab577
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/test/axon/loop_test.exs b/test/axon/loop_test.exs
@@ -854,7 +854,11 @@ defmodule Axon.LoopTest do
     test "saves a checkpoint on custom events", %{loop: loop} do
       data = List.duplicate({Nx.iota({1, 1}), Nx.iota({1, 1})}, 5)
 
-      assert %Axon.Loop.State{epoch: 3, iteration: 0, event_counts: %{iteration_completed: %{total: 15}}} =
+      assert %Axon.Loop.State{
+               epoch: 3,
+               iteration: 0,
+               event_counts: %{iteration_completed: %{total: 15}}
+             } =
                loop
                |> Map.put(:output_transform, & &1)
                |> Loop.checkpoint(event: :iteration_completed, filter: [every: {:epoch, 2}])