enhance ut

Signed-off-by: Mengni Wang <[email protected]>
microsoft · Jun 20, 2023 · 13adeab · 13adeab
1 parent 636ffd5
commit 13adeab
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/onnxruntime/test/python/quantization/test_quantize_static.py b/onnxruntime/test/python/quantization/test_quantize_static.py
@@ -109,7 +109,7 @@ def test_smooth_quant(self):
         data_reader.rewind()
 
         model = onnx.load(quant_model_path)
-        self.assertTrue("Mul" in [i.op_type for i in model.graph.node])
+        self.assertIn("Mul", [i.op_type for i in model.graph.node])
 
 
 if __name__ == "__main__":