[GitHub] iamthebot commented on issue #12285: Change the way NDArrayIter handle the last batch

[GitBox <gi...@apache.org>]

iamthebot commented on issue #12285: Change the way NDArrayIter handle the last batch
URL: https://github.com/apache/incubator-mxnet/pull/12285#issuecomment-420475439
 
 
   This is causing a regression which is breaking most of our test cases using simple iteration over `NDArrayIter`.
   
   ```
   .com/airbnb/bighead/python/bighead/ml_frameworks/mxnet/gluon.py", line 434, in transform
       for batch in data_iter:
     File "/anaconda3/envs/py36/lib/python3.6/site-packages/mxnet/io/io.py", line 228, in __next__
       return self.next()
     File "/anaconda3/envs/py36/lib/python3.6/site-packages/mxnet/io/io.py", line 680, in next
       label = self.getlabel()
     File "/anaconda3/envs/py36/lib/python3.6/site-packages/mxnet/io/io.py", line 750, in getlabel
       return self._batchify(self.label)
     File "/anaconda3/envs/py36/lib/python3.6/site-packages/mxnet/io/io.py", line 732, in _batchify
       first_data = self._getdata(data_source, start=self.cursor)
     File "/anaconda3/envs/py36/lib/python3.6/site-packages/mxnet/io/io.py", line 694, in _getdata
       end = end if end is not None else data_source[0][1].shape[0]
   IndexError: list index out of range
   ```

----------------------------------------------------------------
This is an automated message from the Apache Git Service.
To respond to the message, please log on GitHub and use the
URL above to go to the specific comment.
 
For queries about this service, please contact Infrastructure at:
users@infra.apache.org


With regards,
Apache Git Services