提交 5b452bfd 编写于 作者: M minqiyang

Remove the overfix of lod_rank_table

上级 47561c34
...@@ -18,7 +18,6 @@ from paddle.fluid.executor import Executor ...@@ -18,7 +18,6 @@ from paddle.fluid.executor import Executor
import paddle.fluid.core as core import paddle.fluid.core as core
import numpy import numpy
import unittest import unittest
import six
class TestLoDRankTable(unittest.TestCase): class TestLoDRankTable(unittest.TestCase):
...@@ -37,7 +36,7 @@ class TestLoDRankTable(unittest.TestCase): ...@@ -37,7 +36,7 @@ class TestLoDRankTable(unittest.TestCase):
exe.run(scope=scope, feed={'x': tensor}) exe.run(scope=scope, feed={'x': tensor})
var = scope.find_var(rank_table.name) var = scope.find_var(rank_table.name)
table = var.get_lod_rank_table() 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__': if __name__ == '__main__':
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册