Skip to content

Commit

Permalink
flake8
Browse files Browse the repository at this point in the history
Signed-off-by: Ananya <[email protected]>
  • Loading branch information
ananyamukh6 committed Sep 3, 2023
1 parent a4db367 commit d674903
Showing 1 changed file with 2 additions and 1 deletion.
3 changes: 2 additions & 1 deletion onnxoptimizer/test/optimizer_test.py
Original file line number Diff line number Diff line change
Expand Up @@ -4613,9 +4613,10 @@ def test_eliminate_consecutive_idempotent_sign_op(self):
""")

optimized_model = self._optimized(
model, ['eliminate_consecutive_idempotent_ops', 'eliminate_deadend'], True)
model, ['eliminate_consecutive_idempotent_ops', 'eliminate_deadend'], True)
assert len(optimized_model.graph.node) == 1
assert optimized_model.graph.node[0].op_type == "Sign"


if __name__ == "__main__":
unittest.main()

0 comments on commit d674903

Please sign in to comment.