We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 3fa8be4 commit 00cf665Copy full SHA for 00cf665
onnxoptimizer/test/optimizer_test.py
@@ -4615,14 +4615,13 @@ def test_rewrite_where(self):
4615
""")
4616
4617
optimized_model = self._optimized(
4618
- model,["rewrite_where"], True)
+ model, ["rewrite_where"], True)
4619
4620
assert len(optimized_model.graph.node) == 5
4621
assert set([i.op_type for i in optimized_model.graph.node]) == {'Where', 'And', 'Sign'}
4622
assert optimized_model.graph.node[0].input == ['A', 'Y', 'X']
4623
assert optimized_model.graph.node[3].input == ['M', 'X', 'Y']
4624
4625
4626
-
4627
if __name__ == "__main__":
4628
unittest.main()
0 commit comments