提交 9b522a86 authored 作者: Ricardo Vieira's avatar Ricardo Vieira 提交者: Ricardo Vieira

Suppress numpy warnings in python implementation of Elemwise

上级 d87d44ce
...@@ -707,7 +707,8 @@ class Elemwise(OpenMPOp): ...@@ -707,7 +707,8 @@ class Elemwise(OpenMPOp):
nout = ufunc.nout nout = ufunc.nout
variables = ufunc(*ufunc_args, **ufunc_kwargs) with np.errstate(all="ignore"):
variables = ufunc(*ufunc_args, **ufunc_kwargs)
if nout == 1: if nout == 1:
variables = [variables] variables = [variables]
......
...@@ -1093,3 +1093,11 @@ def test_gradient_mixed_discrete_output_scalar_op(): ...@@ -1093,3 +1093,11 @@ def test_gradient_mixed_discrete_output_scalar_op():
np.ones((12,), dtype=config.floatX), np.ones((12,), dtype=config.floatX),
strict=True, strict=True,
) )
@pytest.mark.filterwarnings("error")
def test_numpy_warning_suppressed():
x = pt.scalar("x")
y = pt.log(x)
fn = pytensor.function([x], y, mode=Mode(linker="py"))
assert fn(0) == -np.inf
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论