Skip to content

Commit 8b5c55e

Browse files
committed
[471] running pre-commit linting
Signed-off-by: Nathaniel <[email protected]>
1 parent d1495a7 commit 8b5c55e

File tree

1 file changed

+4
-2
lines changed

1 file changed

+4
-2
lines changed

pytensor/tensor/rewriting/math.py

Lines changed: 4 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -362,9 +362,11 @@ def local_exp_log_nan_switch(fgraph, node):
362362
old_out = node.outputs[0]
363363
new_out = switch(le(x, 0), sub(1, exp(x)), np.asarray(np.nan, old_out.dtype))
364364
return [new_out]
365-
365+
366366
# Case for log1mexp(log1mexp(x)) -> x
367-
if isinstance(prev_op, aes_math.Log1mexp) and isinstance(node_op, aes_math.Log1mexp):
367+
if isinstance(prev_op, aes_math.Log1mexp) and isinstance(
368+
node_op, aes_math.Log1mexp
369+
):
368370
x = x.owner.inputs[0]
369371
old_out = node.outputs[0]
370372
new_out = switch(le(x, 0), x, np.asarray(np.nan, old_out.dtype))

0 commit comments

Comments
 (0)