Skip to content

Commit 6547245

Browse files
enosairjerryzh168
authored andcommitted
Add return value to setup() function of PipedReaderBuilder (pytorch#7476)
1 parent 6c7a831 commit 6547245

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

caffe2/python/dataio.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -243,7 +243,7 @@ def schema(self):
243243
return self._builder.schema()
244244

245245
def setup(self, **kwargs):
246-
self._builder.setup(**kwargs)
246+
return self._builder.setup(**kwargs)
247247

248248
def new_reader(self, **kwargs):
249249
# Passing everything down since you could wrap a PipedReaderBuilder in

0 commit comments

Comments
 (0)