From 13adeabc2421775a0c7ce70c3939c853823ce15a Mon Sep 17 00:00:00 2001 From: Mengni Wang Date: Tue, 20 Jun 2023 09:38:23 +0800 Subject: [PATCH] enhance ut Signed-off-by: Mengni Wang --- onnxruntime/test/python/quantization/test_quantize_static.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/onnxruntime/test/python/quantization/test_quantize_static.py b/onnxruntime/test/python/quantization/test_quantize_static.py index f771ee3959344..5f91eb68783f9 100644 --- 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__":