Skip to content

Commit 4349e62

Browse files
committed
fix format. test=develop
1 parent 55ed424 commit 4349e62

File tree

2 files changed

+8
-6
lines changed

2 files changed

+8
-6
lines changed

python/paddle/fluid/dataloader/worker.py

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -282,8 +282,9 @@ def _worker_loop(dataset, dataset_kind, indices_queue, out_queue, done_event,
282282
try:
283283
if init_fn is not None:
284284
init_fn(worker_id)
285-
fetcher = _DatasetKind.create_fetcher(
286-
dataset_kind, dataset, auto_collate_batch, collate_fn, drop_last)
285+
fetcher = _DatasetKind.create_fetcher(dataset_kind, dataset,
286+
auto_collate_batch,
287+
collate_fn, drop_last)
287288
except:
288289
init_exception = _WorkerException(worker_id)
289290

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

Lines changed: 5 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -403,10 +403,11 @@ def run_main(self, dataset, num_samples, batch_size):
403403
for drop_last in [True, False]:
404404
steps = (num_samples + (1 - int(drop_last)) * \
405405
(batch_size - 1)) // batch_size
406-
dataloader = DataLoader(dataset,
407-
batch_size=batch_size,
408-
drop_last=drop_last,
409-
num_workers=num_workers)
406+
dataloader = DataLoader(
407+
dataset,
408+
batch_size=batch_size,
409+
drop_last=drop_last,
410+
num_workers=num_workers)
410411
datas = []
411412
for data in dataloader:
412413
datas.append(data)

0 commit comments

Comments
 (0)