--version

polvalente · polvalente · commit 924a1c03c472 · 2025-05-06T16:27:03.000-03:00
diff --git a/nx/lib/nx/testing.ex b/nx/lib/nx/testing.ex
@@ -25,7 +25,7 @@ defmodule Nx.Testing do
   end
 
   def assert_equal(left, right) do
-    if !tensor_equal?(left, right) do
+    if not tensor_equal?(left, right) do
       flunk("""
       Tensor assertion failed.
       left: #{inspect(left)}
@@ -34,7 +34,7 @@ defmodule Nx.Testing do
     end
   end
 
-  def tensor_equal?(left, right) do
+  defp tensor_equal?(left, right) do
     both_nan = Nx.is_nan(left) |> Nx.logical_and(Nx.is_nan(right))
 
     left
