Skip to content

Commit 2524024

Browse files
committed
flake8
Signed-off-by: Ananya <[email protected]>
1 parent 0880593 commit 2524024

File tree

1 file changed

+1
-2
lines changed

1 file changed

+1
-2
lines changed

onnxoptimizer/test/optimizer_test.py

+1-2
Original file line numberDiff line numberDiff line change
@@ -4615,14 +4615,13 @@ def test_rewrite_where(self):
46154615
""")
46164616

46174617
optimized_model = self._optimized(
4618-
model,["rewrite_where"], True)
4618+
model, ["rewrite_where"], True)
46194619

46204620
assert len(optimized_model.graph.node) == 5
46214621
assert set([i.op_type for i in optimized_model.graph.node]) == {'Where', 'And', 'Sign'}
46224622
assert optimized_model.graph.node[0].input == ['A', 'Y', 'X']
46234623
assert optimized_model.graph.node[3].input == ['M', 'X', 'Y']
46244624

46254625

4626-
46274626
if __name__ == "__main__":
46284627
unittest.main()

0 commit comments

Comments
 (0)