From 25240248104b95d940a061de76969370fc432a77 Mon Sep 17 00:00:00 2001 From: Ananya Date: Tue, 29 Aug 2023 21:51:07 -0700 Subject: [PATCH] flake8 Signed-off-by: Ananya --- onnxoptimizer/test/optimizer_test.py | 3 +-- 1 file changed, 1 insertion(+), 2 deletions(-) diff --git a/onnxoptimizer/test/optimizer_test.py b/onnxoptimizer/test/optimizer_test.py index e591b41ae..3c8107a4b 100644 --- a/onnxoptimizer/test/optimizer_test.py +++ b/onnxoptimizer/test/optimizer_test.py @@ -4615,7 +4615,7 @@ def test_rewrite_where(self): """) optimized_model = self._optimized( - model,["rewrite_where"], True) + model, ["rewrite_where"], True) assert len(optimized_model.graph.node) == 5 assert set([i.op_type for i in optimized_model.graph.node]) == {'Where', 'And', 'Sign'} @@ -4623,6 +4623,5 @@ def test_rewrite_where(self): assert optimized_model.graph.node[3].input == ['M', 'X', 'Y'] - if __name__ == "__main__": unittest.main()