Fix obsured warning (#1944)

2025-09-19 02:38:09 +08:00 · 2025-03-09 06:50:39 +08:00
parent d699cc1330
commit 5db90ce822
1 changed files with 1 additions and 3 deletions
--- 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)