This is an automated email from the ASF dual-hosted git repository. zhasheng pushed a commit to branch master in repository https://gitbox.apache.org/repos/asf/incubator-mxnet.git
The following commit(s) were added to refs/heads/master by this push: new fe5b56e Fix DataBatch.__str__ for cases where we don't have labels. (#9645) fe5b56e is described below commit fe5b56e419d454dc8f42f0307f53ced133804ca7 Author: Pedro Larroy <928489+lar...@users.noreply.github.com> AuthorDate: Sat Feb 3 08:25:20 2018 +0100 Fix DataBatch.__str__ for cases where we don't have labels. (#9645) --- python/mxnet/io.py | 5 ++++- tests/python/unittest/test_io.py | 11 +++++++++++ 2 files changed, 15 insertions(+), 1 deletion(-) diff --git a/python/mxnet/io.py b/python/mxnet/io.py index b07f7c1..201414e 100644 --- a/python/mxnet/io.py +++ b/python/mxnet/io.py @@ -168,7 +168,10 @@ class DataBatch(object): def __str__(self): data_shapes = [d.shape for d in self.data] - label_shapes = [l.shape for l in self.label] + if self.label: + label_shapes = [l.shape for l in self.label] + else: + label_shapes = None return "{}: data shapes: {} label shapes: {}".format( self.__class__.__name__, data_shapes, diff --git a/tests/python/unittest/test_io.py b/tests/python/unittest/test_io.py index e8aba38..58ca1d7 100644 --- a/tests/python/unittest/test_io.py +++ b/tests/python/unittest/test_io.py @@ -252,6 +252,17 @@ def test_LibSVMIter(): check_libSVMIter_synthetic() check_libSVMIter_news_data() + +def test_DataBatch(): + from nose.tools import ok_ + from mxnet.io import DataBatch + import re + batch = DataBatch(data=[mx.nd.ones((2,3))]) + ok_(re.match('DataBatch: data shapes: \[\(2L?, 3L?\)\] label shapes: None', str(batch))) + batch = DataBatch(data=[mx.nd.ones((2,3)), mx.nd.ones((7,8))], label=[mx.nd.ones((4,5))]) + ok_(re.match('DataBatch: data shapes: \[\(2L?, 3L?\), \(7L?, 8L?\)\] label shapes: \[\(4L?, 5L?\)\]', str(batch))) + + @unittest.skip("test fails intermittently. temporarily disabled till it gets fixed. tracked at https://github.com/apache/incubator-mxnet/issues/7826") def test_CSVIter(): def check_CSVIter_synthetic(): -- To stop receiving notification emails like this one, please contact zhash...@apache.org.