From 514888356ea12232f7aad7f64d17a2972ced5398 Mon Sep 17 00:00:00 2001
From: pengjuncai <13006307475@163.com>
Date: Tue, 15 Jun 2021 07:55:40 +0000
Subject: [PATCH] fix the error of qat unit test

---
 .../contrib/slim/tests/test_imperative_qat.py    | 16 +++-------------
 .../tests/test_imperative_qat_channelwise.py     |  3 ---
 2 files changed, 3 insertions(+), 16 deletions(-)

diff --git a/python/paddle/fluid/contrib/slim/tests/test_imperative_qat.py b/python/paddle/fluid/contrib/slim/tests/test_imperative_qat.py
index bf411e5b38efa..3cc61ce8c5808 100644
--- a/python/paddle/fluid/contrib/slim/tests/test_imperative_qat.py
+++ b/python/paddle/fluid/contrib/slim/tests/test_imperative_qat.py
@@ -64,11 +64,11 @@ def tearDownClass(cls):
             print("Failed to delete {} due to {}".format(cls.root_path, str(e)))
 
     def set_vars(self):
-        self.weight_quantize_type = None
-        self.activation_quantize_type = None
+        self.weight_quantize_type = 'abs_max'
+        self.activation_quantize_type = 'moving_average_abs_max'
         print('weight_quantize_type', self.weight_quantize_type)
 
-    def run_qat_save(self):
+    def test_qat(self):
         self.set_vars()
 
         imperative_qat = ImperativeQuantAware(
@@ -200,15 +200,5 @@ def run_qat_save(self):
             msg='Failed to save the inference quantized model.')
 
 
-class TestImperativeQatAbsMax(TestImperativeQat):
-    def set_vars(self):
-        self.weight_quantize_type = 'abs_max'
-        self.activation_quantize_type = 'moving_average_abs_max'
-        print('weight_quantize_type', self.weight_quantize_type)
-
-    def test_qat(self):
-        self.run_qat_save()
-
-
 if __name__ == '__main__':
     unittest.main()
diff --git a/python/paddle/fluid/contrib/slim/tests/test_imperative_qat_channelwise.py b/python/paddle/fluid/contrib/slim/tests/test_imperative_qat_channelwise.py
index 3d2cad388d172..1a6c9c41638db 100644
--- a/python/paddle/fluid/contrib/slim/tests/test_imperative_qat_channelwise.py
+++ b/python/paddle/fluid/contrib/slim/tests/test_imperative_qat_channelwise.py
@@ -43,9 +43,6 @@ def set_vars(self):
         self.activation_quantize_type = 'moving_average_abs_max'
         print('weight_quantize_type', self.weight_quantize_type)
 
-    def test_qat(self):
-        self.run_qat_save()
-
 
 if __name__ == '__main__':
     unittest.main()