提交 cbd662fc authored 作者: Virgile Andreani's avatar Virgile Andreani 提交者: Virgile Andreani

Apply rule RUF040

上级 ffd7a5aa
......@@ -107,9 +107,8 @@ def test_blockwise_alloc():
y = tensor("y", shape=())
out = vector_add(alloc(x, 3, 1, 5), alloc(y, 7, 5))
expected_out = alloc(vector_add(alloc(x, 5), alloc(y, 5)), 3, 7, 5)
assert equal([rewrite(out)], [expected_out]), (
None
) # pytensor.dprint([expected_out, rewrite(out)], print_type=True)
# pytensor.dprint([expected_out, rewrite(out)], print_type=True)
assert equal([rewrite(out)], [expected_out])
x = tensor("x", shape=(5,))
y = tensor("y", shape=())
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论