Format

Author: jonatanklosko

test/bumblebee/multimodal/blip_test.exs

@@ -25,7 +25,9 @@ defmodule Bumblebee.Multimodal.BlipTest do
 
     assert_all_close(
       outputs.logits[[.., 1..3, 1..3]],
-      Nx.tensor([[[0.1215, 0.0226, -0.1134], [0.1472, 0.1118, 0.1031], [-0.0687, 0.0104, 0.1781]]])
+      Nx.tensor([
+        [[0.1215, 0.0226, -0.1134], [0.1472, 0.1118, 0.1031], [-0.0687, 0.0104, 0.1781]]
+      ])
     )
   end
 end

test/bumblebee/multimodal/layout_lm_test.exs

@@ -37,7 +37,9 @@ defmodule Bumblebee.Multimodal.LayoutLmTest do
 
     assert_all_close(
       outputs.hidden_state[[.., 1..3, 1..3]],
-      Nx.tensor([[[0.0240, -0.8855, 1.8877], [1.8435, 0.6223, 2.0573], [1.6961, -1.2411, 1.2824]]])
+      Nx.tensor([
+        [[0.0240, -0.8855, 1.8877], [1.8435, 0.6223, 2.0573], [1.6961, -1.2411, 1.2824]]
+      ])
     )
   end
 

test/bumblebee/text/bert_test.exs

@@ -45,7 +45,9 @@ defmodule Bumblebee.Text.BertTest do
 
     assert_all_close(
       outputs.logits[[.., 1..3, 1..3]],
-      Nx.tensor([[[-0.0127, 0.0508, 0.0904], [0.1151, 0.1189, 0.0922], [0.0089, 0.1132, -0.2470]]])
+      Nx.tensor([
+        [[-0.0127, 0.0508, 0.0904], [0.1151, 0.1189, 0.0922], [0.0089, 0.1132, -0.2470]]
+      ])
     )
   end
 

test/bumblebee/text/gpt2_test.exs

@@ -45,7 +45,9 @@ defmodule Bumblebee.Text.Gpt2Test do
 
     assert_all_close(
       outputs.logits[[.., 1..3, 1..3]],
-      Nx.tensor([[[0.1184, -0.0259, 0.1688], [0.1064, 0.1412, 0.1120], [0.1421, -0.2010, 0.3757]]])
+      Nx.tensor([
+        [[0.1184, -0.0259, 0.1688], [0.1064, 0.1412, 0.1120], [0.1421, -0.2010, 0.3757]]
+      ])
     )
   end
 

test/bumblebee/text/phi_test.exs

@@ -22,7 +22,9 @@ defmodule Bumblebee.Text.PhiTest do
 
     assert_all_close(
       outputs.hidden_state[[.., 1..3, 1..3]],
-      Nx.tensor([[[-0.3275, 0.5231, 0.5690], [0.2239, 0.5028, 0.4599], [-0.0979, 1.0183, 0.3350]]])
+      Nx.tensor([
+        [[-0.3275, 0.5231, 0.5690], [0.2239, 0.5028, 0.4599], [-0.0979, 1.0183, 0.3350]]
+      ])
     )
   end
 

test/bumblebee/text/roberta_test.exs

@@ -49,7 +49,9 @@ defmodule Bumblebee.Text.RobertaTest do
 
     assert_all_close(
       outputs.logits[[.., 1..3, 1..3]],
-      Nx.tensor([[[0.0000, -0.0796, 0.1734], [0.0000, -0.0754, 0.0755], [0.0000, 0.0299, 0.1902]]])
+      Nx.tensor([
+        [[0.0000, -0.0796, 0.1734], [0.0000, -0.0754, 0.0755], [0.0000, 0.0299, 0.1902]]
+      ])
     )
   end
 
@@ -176,7 +178,9 @@ defmodule Bumblebee.Text.RobertaTest do
 
     assert_all_close(
       outputs.logits[[.., 1..3, 1..3]],
-      Nx.tensor([[[0.0000, 0.0661, -0.0063], [0.0000, 0.1107, -0.1137], [0.0000, 0.1044, 0.0803]]])
+      Nx.tensor([
+        [[0.0000, 0.0661, -0.0063], [0.0000, 0.1107, -0.1137], [0.0000, 0.1044, 0.0803]]
+      ])
     )
   end
 end

test/bumblebee/text/t5_test.exs

@@ -82,7 +82,9 @@ defmodule Bumblebee.Text.T5Test do
 
     assert_all_close(
       outputs.logits[[.., 1..3, 1..3]] |> Nx.multiply(10_000),
-      Nx.tensor([[[-0.0158, 0.0067, 0.0636], [0.0128, 0.0742, -0.0398], [0.0050, 0.0554, 0.0083]]])
+      Nx.tensor([
+        [[-0.0158, 0.0067, 0.0636], [0.0128, 0.0742, -0.0398], [0.0050, 0.0554, 0.0083]]
+      ])
     )
   end
 

test/bumblebee/vision/convnext_test.exs

@@ -22,7 +22,9 @@ defmodule Bumblebee.Vision.ConvNextTest do
 
     assert_all_close(
       outputs.hidden_state[[.., 1..2, 1..2, 1..2]],
-      Nx.tensor([[[[0.3924, -0.2330], [0.3924, -0.2330]], [[0.3924, -0.2330], [0.3924, -0.2330]]]])
+      Nx.tensor([
+        [[[0.3924, -0.2330], [0.3924, -0.2330]], [[0.3924, -0.2330], [0.3924, -0.2330]]]
+      ])
     )
 
     assert_all_close(

test/bumblebee/vision/vit_test.exs

@@ -73,7 +73,9 @@ defmodule Bumblebee.Vision.VitTest do
 
     assert_all_close(
       outputs.pixel_values[[.., 1..2, 1..2, 1..2]],
-      Nx.tensor([[[[0.0752, 0.0548], [-0.0192, -0.0216]], [[-0.0252, 0.0728], [0.0232, -0.1687]]]])
+      Nx.tensor([
+        [[[0.0752, 0.0548], [-0.0192, -0.0216]], [[-0.0252, 0.0728], [0.0232, -0.1687]]]
+      ])
     )
   end
 end