From b8b61f5bbe588a8c41aa8eeec28eb3377c70a0ed Mon Sep 17 00:00:00 2001 From: Neil Girdhar Date: Sat, 14 Sep 2024 15:59:03 -0400 Subject: [PATCH] Polish math-tools overloads --- tjax/_src/math_tools.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tjax/_src/math_tools.py b/tjax/_src/math_tools.py index 0fe9981f..fea19d2b 100644 --- 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))