Skip to content

Commit 48b088c

Browse files
committed
mix format
1 parent 7549686 commit 48b088c

File tree

1 file changed

+9
-1
lines changed

1 file changed

+9
-1
lines changed

exla/test/exla/defn/vectorize_test.exs

Lines changed: 9 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -333,7 +333,15 @@ defmodule EXLA.Defn.VectorizeTest do
333333

334334
test "2 vectorized preds with different axes" do
335335
assert_equal(
336-
cond4(Nx.vectorize(~VEC[0 1 0], :pred1), 10, Nx.vectorize(~VEC[1 0], :pred2), 20, 0, 30, 40),
336+
cond4(
337+
Nx.vectorize(~VEC[0 1 0], :pred1),
338+
10,
339+
Nx.vectorize(~VEC[1 0], :pred2),
340+
20,
341+
0,
342+
30,
343+
40
344+
),
337345
Nx.vectorize(~MAT[
338346
20 40
339347
10 10

0 commit comments

Comments
 (0)