fix ut

7 years ago · 69188e5981
parent c6d7c2bdff
commit 69188e5981
1 changed files with 1 additions and 0 deletions
--- a/python/paddle/fluid/tests/unittests/test_split_op.py
+++ b/python/paddle/fluid/tests/unittests/test_split_op.py
@ -19,6 +19,7 @@ from op_test import OpTest

 class TestSplitOp(OpTest):
    def setUp(self):
+        self._set_op_type()
        axis = 1
        x = np.random.random((4, 5, 6)).astype('float32')
        out = np.split(x, [2, 3], axis)