diff --git a/python/paddle/v2/reader/creator.py b/python/paddle/v2/reader/creator.py index 994062096fbe910c8189d3ecd49298781dec708f..8888faae36ad86c4af0234f4110eae20ca3a5831 100644 --- a/python/paddle/v2/reader/creator.py +++ b/python/paddle/v2/reader/creator.py @@ -57,15 +57,17 @@ def text_file(path): return reader -def RecordIO(path): +def recordio(path): """ Creates a data reader that outputs record one one by one from given recordio file :path: path of recordio file :returns: data reader of recordio file """ + import recordio as rec + def reader(): - f = recordio.reader(path) + f = rec.reader(path) while True: r = f.read() if r is None: diff --git a/python/paddle/v2/reader/tests/creator_test.py b/python/paddle/v2/reader/tests/creator_test.py index dd84fbb002543f2c3d90c9543e475516165a5f4d..e20af9e5e4c70c0c273a622f3e5ad22fba5c273f 100644 --- a/python/paddle/v2/reader/tests/creator_test.py +++ b/python/paddle/v2/reader/tests/creator_test.py @@ -37,10 +37,10 @@ class TestTextFile(unittest.TestCase): class TestRecordIO(unittest.TestCase): - def test_RecordIO(self): + def test_recordio(self): path = os.path.join( os.path.dirname(__file__), "test_recordio_creator.dat") - reader = paddle.v2.reader.creator.RecordIO(path) + reader = paddle.v2.reader.creator.recordio(path) for idx, r in enumerate(reader()): self.assertSequenceEqual(r, str(idx))