flake8

Signed-off-by: Ananya <[email protected]>
onnx · Sep 3, 2023 · 2524024 · 2524024
1 parent 0880593
commit 2524024
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/onnxoptimizer/test/optimizer_test.py b/onnxoptimizer/test/optimizer_test.py
@@ -4615,14 +4615,13 @@ 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'}
         assert optimized_model.graph.node[0].input == ['A', 'Y', 'X']
         assert optimized_model.graph.node[3].input == ['M', 'X', 'Y']
 
 
-
 if __name__ == "__main__":
     unittest.main()