chore: format

Author: polvalente

test/axon/serialization_guide_test.exs

@@ -5,7 +5,10 @@ defmodule Axon.SerializationGuideTest do
   """
   use Axon.Case, async: false
 
-  @tmp_path Path.join(System.tmp_dir!(), "axon_serialization_guide_test_#{:erlang.unique_integer([:positive])}")
+  @tmp_path Path.join(
+              System.tmp_dir!(),
+              "axon_serialization_guide_test_#{:erlang.unique_integer([:positive])}"
+            )
 
   setup do
     File.mkdir_p!(@tmp_path)
@@ -28,12 +31,18 @@ defmodule Axon.SerializationGuideTest do
 
       train_data =
         Stream.repeatedly(fn ->
-          {xs, _} = Nx.Random.normal(Nx.Random.key(:erlang.phash2({self(), System.unique_integer([:monotonic])})), shape: {8, 1})
+          {xs, _} =
+            Nx.Random.normal(
+              Nx.Random.key(:erlang.phash2({self(), System.unique_integer([:monotonic])})),
+              shape: {8, 1}
+            )
+
           {xs, Nx.sin(xs)}
         end)
 
       # Train
-      trained_model_state = Axon.Loop.run(loop, train_data, Axon.ModelState.empty(), epochs: 2, iterations: 50)
+      trained_model_state =
+        Axon.Loop.run(loop, train_data, Axon.ModelState.empty(), epochs: 2, iterations: 50)
 
       # Extract and save params (as in guide)
       params =
@@ -120,7 +129,10 @@ defmodule Axon.SerializationGuideTest do
       [ckpt_file] = File.ls!(checkpoint_path)
       state = File.read!(Path.join(checkpoint_path, ckpt_file)) |> Axon.Loop.deserialize_state()
 
-      resumed_loop = model |> Axon.Loop.trainer(:mean_squared_error, :sgd, log: 0) |> Axon.Loop.from_state(state)
+      resumed_loop =
+        model
+        |> Axon.Loop.trainer(:mean_squared_error, :sgd, log: 0)
+        |> Axon.Loop.from_state(state)
 
       # Resume - should complete without error
       result = Axon.Loop.run(resumed_loop, train_data, Axon.ModelState.empty(), epochs: 2)