Update tests/trainer/test_dataloaders.py

awaelchli · web-flow · commit 9727562a7642 · 2020-10-22T08:15:45.000+02:00
diff --git a/tests/trainer/test_dataloaders.py b/tests/trainer/test_dataloaders.py
@@ -771,7 +771,7 @@ def train_dataloader(self):
         distributed_backend='ddp_spawn',
         default_root_dir=tmpdir,
         max_steps=100,
-        callbacks=[DistribSamplerCallback()]
+        callbacks=[DistribSamplerCallback()],
         replace_sampler_ddp=True,
     )
     trainer.fit(model)

Original file line number	Diff line number	Diff line change
`@@ -771,7 +771,7 @@ def train_dataloader(self):`
`771`	`771`	`distributed_backend='ddp_spawn',`
`772`	`772`	`default_root_dir=tmpdir,`
`773`	`773`	`max_steps=100,`
`774`		`- callbacks=[DistribSamplerCallback()]`
	`774`	`+ callbacks=[DistribSamplerCallback()],`
`775`	`775`	`replace_sampler_ddp=True,`
`776`	`776`	`)`
`777`	`777`	`trainer.fit(model)`