From df5bc55410b7cb9de1d98fd17a8c4bbb92f5a0e1 Mon Sep 17 00:00:00 2001 From: adrianlizarraga Date: Tue, 30 Jan 2024 12:51:57 -0800 Subject: [PATCH] Use assertIsNot() instead of assertTrue(a is not None) --- onnxruntime/test/python/quantization/test_qdq.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/onnxruntime/test/python/quantization/test_qdq.py b/onnxruntime/test/python/quantization/test_qdq.py index 8504f0f90fa89..223f405e8947a 100644 --- a/onnxruntime/test/python/quantization/test_qdq.py +++ b/onnxruntime/test/python/quantization/test_qdq.py @@ -606,7 +606,7 @@ def verify_qdq(self, per_channel, activation_type, weight_type, extra_options=No if extra_options and extra_options.get("UseQDQContribOps", False): qdq_model = onnx.load_model(model_qdq_path) ms_opset = next((opset for opset in qdq_model.opset_import if opset.domain == "com.microsoft"), None) - self.assertTrue(ms_opset is not None) + self.assertIsNot(ms_opset, None) def verify_qop(self, per_channel, is_quant_type_int8): np.random.seed(1)