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__":