mix format

elixir-nx · May 13, 2024 · 48b088c · 48b088c
1 parent 7549686
commit 48b088c
Showing 1 changed file with 9 additions and 1 deletion.
diff --git a/exla/test/exla/defn/vectorize_test.exs b/exla/test/exla/defn/vectorize_test.exs
@@ -333,7 +333,15 @@ defmodule EXLA.Defn.VectorizeTest do
 
     test "2 vectorized preds with different axes" do
       assert_equal(
-        cond4(Nx.vectorize(~VEC[0 1 0], :pred1), 10, Nx.vectorize(~VEC[1 0], :pred2), 20, 0, 30, 40),
+        cond4(
+          Nx.vectorize(~VEC[0 1 0], :pred1),
+          10,
+          Nx.vectorize(~VEC[1 0], :pred2),
+          20,
+          0,
+          30,
+          40
+        ),
         Nx.vectorize(~MAT[
               20 40
               10 10