Skip to content

Commit 55ed424

Browse files
committed
fix unittests. test=develop
1 parent c5c8c64 commit 55ed424

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

python/paddle/fluid/tests/unittests/test_multiprocess_dataloader_exception.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -180,7 +180,7 @@ def _collate_fn(sample_list):
180180
indices_queue.put(None)
181181
_worker_loop(loader._dataset, 0, indices_queue,
182182
loader._data_queue, loader._workers_done_event,
183-
True, _collate_fn, _init_fn, 0, 1,
183+
True, _collate_fn, True, _init_fn, 0, 1,
184184
loader._use_shared_memory)
185185
self.assertTrue(False)
186186
except AssertionError:
@@ -224,7 +224,7 @@ def _collate_fn(sample_list):
224224
loader._workers_done_event.set()
225225
_worker_loop(loader._dataset, 0, indices_queue,
226226
loader._data_queue, loader._workers_done_event,
227-
True, _collate_fn, _init_fn, 0, 1,
227+
True, _collate_fn, True, _init_fn, 0, 1,
228228
loader._use_shared_memory)
229229
self.assertTrue(True)
230230
except AssertionError:

0 commit comments

Comments
 (0)