Formatting

elixir-nx · Oct 21, 2024 · 4a66ba4 · 4a66ba4
1 parent a8d3149
commit 4a66ba4
Showing 1 changed file with 4 additions and 3 deletions.
diff --git a/test/axon/compiler_test.exs b/test/axon/compiler_test.exs
@@ -136,9 +136,10 @@ defmodule CompilerTest do
 
       {_, predict_fn} = Axon.build(model)
 
-      exception = assert_raise ArgumentError, fn ->
-        predict_fn.(ModelState.empty(), Nx.tensor([1]))
-      end
+      exception =
+        assert_raise ArgumentError, fn ->
+          predict_fn.(ModelState.empty(), Nx.tensor([1]))
+        end
 
       assert Exception.message(exception) =~ "ambiguous"
     end