Update onnxscript/rewriter/generic_pattern_test.py

Co-authored-by: Justin Chu <[email protected]>
microsoft · Jul 16, 2024 · 8f5878e · 8f5878e
1 parent d80c578
commit 8f5878e
Showing 1 changed file with 2 additions and 3 deletions.
diff --git a/onnxscript/rewriter/generic_pattern_test.py b/onnxscript/rewriter/generic_pattern_test.py
@@ -312,9 +312,8 @@ def apply_pattern(op, x, **_):
         model = onnx.shape_inference.infer_shapes(model_proto)
         ir_model = ir.serde.deserialize_model(model)
         rule.apply_to_model(ir_model)
-        rewritten_model = ir.serde.serialize_model(ir_model)
-        graph = rewritten_model.graph
-        self.assertEqual(len(graph.node), 3)
+        graph = ir_model.graph
+        self.assertEqual(len(graph), 3)
         self.assertEqual(graph.node[0].op_type, "SinCos")
 
     def test_rotary_embedding(self):