Polish math-tools overloads

NeilGirdhar · Sep 23, 2024 · b8b61f5 · b8b61f5
1 parent 09ad5b3
commit b8b61f5
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tjax/_src/math_tools.py b/tjax/_src/math_tools.py
@@ -122,7 +122,7 @@ def divide_nonnegative(dividend: RealArray, divisor: RealArray) -> RealArray:
     whenever the divisor equals zero.
     """
     xp = get_namespace(dividend, divisor)
-    return divide_where(dividend, divisor, where=divisor > 0.0,  # pyright: ignore
+    return divide_where(dividend, divisor, where=divisor > 0.0,  # type: ignore # pyright: ignore
                         otherwise=xp.asarray(xp.inf))