Fix unit test in Python 3.5

6 years ago · 49c672d037
parent 605907fe85
commit 49c672d037
1 changed files with 1 additions and 1 deletions
--- a/python/paddle/fluid/contrib/tests/test_quantize_transpiler.py
+++ b/python/paddle/fluid/contrib/tests/test_quantize_transpiler.py
@ -216,7 +216,7 @@ class TestQuantizeTranspiler(unittest.TestCase):
                                 fetch_list=[loss])

        with fluid.program_guard(test_program):
-            test_data = test_reader().next()
+            test_data = next(test_reader())
            f_var = fluid.framework.get_var('conv2d_1.tmp_0', test_program)
            w_var = fluid.framework.get_var('conv2d_1.w_0.quantized',
                                            test_program)