nit

Author: crusaderky

pixi.lock

@@ -238,6 +238,8 @@ reportUnreachable = false
 reportUnusedParameter = false
 # cyclic imports inside function bodies
 reportImportCycles = false
+# PyRight can't trace types in lambdas
+reportUnknownLambdaType = false
 
 executionEnvironments = [
     { root = "tests", reportPrivateUsage = false },

pyproject.toml

@@ -567,23 +567,15 @@ def isclose(
     b_inexact = xp.isdtype(b.dtype, ("real floating", "complex floating"))
     if a_inexact or b_inexact:
         # prevent warnings on numpy and dask on inf - inf
-        meta_xp = meta_namespace(a, b, xp=xp)
-
-        def where_inf(a: Array, b: Array) -> Array:
-            return (
-                meta_xp.isinf(a)
-                & meta_xp.isinf(b)
-                & (meta_xp.sign(a) == meta_xp.sign(b))
-            )
-
-        def where_not_inf(a: Array, b: Array) -> Array:
-            # Note: inf <= inf is True!
-            return meta_xp.abs(a - b) <= (atol + rtol * meta_xp.abs(b))
-
+        mxp = meta_namespace(a, b, xp=xp)
         out = apply_where(
-            xp.isinf(a) | xp.isinf(b), where_inf, where_not_inf, a, b, xp=xp
+            xp.isinf(a) | xp.isinf(b),
+            lambda a, b: mxp.isinf(a) & mxp.isinf(b) & (mxp.sign(a) == mxp.sign(b)),  # pyright: ignore[reportUnknownArgumentType]
+            # Note: inf <= inf is True!
+            lambda a, b: mxp.abs(a - b) <= (atol + rtol * mxp.abs(b)),  # pyright: ignore[reportUnknownArgumentType]
+            *(a, b),
+            xp=xp,
         )
-
         if equal_nan:
             out = xp.where(xp.isnan(a) & xp.isnan(b), xp.asarray(True), out)
         return out