From 5b452bfd59461a6f9282b17ca9762f39a2b3eb17 Mon Sep 17 00:00:00 2001 From: minqiyang Date: Mon, 13 Aug 2018 16:14:46 +0800 Subject: [PATCH] Remove the overfix of lod_rank_table --- python/paddle/fluid/tests/unittests/test_lod_rank_table.py | 3 +-- 1 file changed, 1 insertion(+), 2 deletions(-) diff --git a/python/paddle/fluid/tests/unittests/test_lod_rank_table.py b/python/paddle/fluid/tests/unittests/test_lod_rank_table.py index cae8f3fb81..d53ead381d 100644 --- a/python/paddle/fluid/tests/unittests/test_lod_rank_table.py +++ b/python/paddle/fluid/tests/unittests/test_lod_rank_table.py @@ -18,7 +18,6 @@ from paddle.fluid.executor import Executor import paddle.fluid.core as core import numpy import unittest -import six class TestLoDRankTable(unittest.TestCase): @@ -37,7 +36,7 @@ class TestLoDRankTable(unittest.TestCase): exe.run(scope=scope, feed={'x': tensor}) var = scope.find_var(rank_table.name) table = var.get_lod_rank_table() - self.assertEqual([(0, 5), (1, 1), (2, 1)], list(six.iteritems(table))) + self.assertEqual([(0, 5), (1, 1), (2, 1)], list(table.items())) if __name__ == '__main__': -- GitLab