Skip to content

Commit 7db4e5e

Browse files
[pre-commit.ci] auto fixes from pre-commit.com hooks
for more information, see https://pre-commit.ci
1 parent 296d545 commit 7db4e5e

File tree

2 files changed

+6
-5
lines changed

2 files changed

+6
-5
lines changed

src/lightning/pytorch/utilities/data.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -169,9 +169,9 @@ def _get_dataloader_init_args_and_kwargs(
169169
if was_wrapped:
170170
# if the dataloader was wrapped in a hook, only take arguments with default values
171171
# and assume user passes their kwargs correctly
172-
params.update(
173-
{k: v for k, v in inspect.signature(DataLoader.__init__).parameters.items() if v.default is not v.empty}
174-
)
172+
params.update({
173+
k: v for k, v in inspect.signature(DataLoader.__init__).parameters.items() if v.default is not v.empty
174+
})
175175
else:
176176
params.update(inspect.signature(DataLoader.__init__).parameters)
177177
params.pop("self", None)

tests/tests_pytorch/trainer/test_batch_sampler.py

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1,12 +1,13 @@
1-
from torch.utils.data import RandomSampler, BatchSampler
1+
from torch.utils.data import BatchSampler, RandomSampler
22
from torch.utils.data.dataloader import DataLoader
33
from torch.utils.data.distributed import DistributedSampler
4+
45
from lightning.pytorch import Callback, Trainer, seed_everything
5-
from tests_pytorch.helpers.runif import RunIf
66
from lightning.pytorch.demos.boring_classes import (
77
BoringModel,
88
RandomDataset,
99
)
10+
from tests_pytorch.helpers.runif import RunIf
1011

1112

1213
class DistribBatchSamplerCallback(Callback):

0 commit comments

Comments
 (0)