From 8d94bc0980116e1f789b1fcbd80688b9a26e7a38 Mon Sep 17 00:00:00 2001
From: Paulo Valente <16843419+polvalente@users.noreply.github.com>
Date: Tue, 24 Sep 2024 17:52:23 -0300
Subject: [PATCH] chore: format

---
 nx/lib/nx/defn/grad.ex | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/nx/lib/nx/defn/grad.ex b/nx/lib/nx/defn/grad.ex
index fcef50b0d6..33a7a0deed 100644
--- a/nx/lib/nx/defn/grad.ex
+++ b/nx/lib/nx/defn/grad.ex
@@ -272,7 +272,7 @@ defmodule Nx.Defn.Grad do
          %{vectorized_axes: vectorized_axes, names: names},
          parent_names
        ) do
-    Keyword.keys(vectorized_axes) ++ Enum.filter(names, & &1 in parent_names)
+    Keyword.keys(vectorized_axes) ++ Enum.filter(names, &(&1 in parent_names))
   end
 
   defp revectorize_node(node, vectorized_names) do