File tree Expand file tree Collapse file tree 2 files changed +6
-5
lines changed
src/lightning/pytorch/utilities
tests/tests_pytorch/trainer Expand file tree Collapse file tree 2 files changed +6
-5
lines changed Original file line number Diff line number Diff 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 )
Original file line number Diff line number Diff line change 1- from torch .utils .data import RandomSampler , BatchSampler
1+ from torch .utils .data import BatchSampler , RandomSampler
22from torch .utils .data .dataloader import DataLoader
33from torch .utils .data .distributed import DistributedSampler
4+
45from lightning .pytorch import Callback , Trainer , seed_everything
5- from tests_pytorch .helpers .runif import RunIf
66from lightning .pytorch .demos .boring_classes import (
77 BoringModel ,
88 RandomDataset ,
99)
10+ from tests_pytorch .helpers .runif import RunIf
1011
1112
1213class DistribBatchSamplerCallback (Callback ):
You can’t perform that action at this time.
0 commit comments