unit test

apache · Jun 3, 2019 · 9762ab2 · 9762ab2
1 parent f679c64
commit 9762ab2
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/python/mxnet/symbol/numpy/_symbol.py b/python/mxnet/symbol/numpy/_symbol.py
@@ -1180,7 +1180,8 @@ def split(ary, indices_or_sections, axis=0):
  indices = [0] + list(indices_or_sections)
  else:
  raise ValueError('indices_or_sections must either int or tuple of ints')
- return _npi.split(ary, indices, axis, sections)
+ ret = _npi.split(ary, indices, axis, False, sections)
+ return ret
 
 
 _set_np_symbol_class(_Symbol)