From 5db90ce8223eab7f09dde21eca4120c2dec0ff21 Mon Sep 17 00:00:00 2001
From: Chunyang Wen <chunyang.wen@gmail.com>
Date: Sun, 9 Mar 2025 06:50:39 +0800
Subject: [PATCH] Fix obsured warning (#1944)

---
 python/mlx/nn/layers/base.py | 4 +---
 1 file changed, 1 insertion(+), 3 deletions(-)

diff --git a/python/mlx/nn/layers/base.py b/python/mlx/nn/layers/base.py
index f141cfc0f..f24bd1806 100644
--- a/python/mlx/nn/layers/base.py
+++ b/python/mlx/nn/layers/base.py
@@ -598,9 +598,7 @@ class Module(dict):
               parameters to the new dtype.
         """
         if predicate is None:
-
-            def predicate(_):
-                return True
+            predicate = lambda _: True
 
         self.apply(lambda x: x.astype(dtype) if predicate(x.dtype) else x)