re-trigger build (#2988)

* fixed build * fixed build
Lightning-AI · Aug 16, 2020 · 766d0f3 · 766d0f3
1 parent 9f2b29a
commit 766d0f3
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 2 deletions.
diff --git a/tests/base/datamodules.py b/tests/base/datamodules.py
@@ -25,7 +25,7 @@ def setup(self, stage: str = None):
             self.dims = self.mnist_train[0][0].shape
 
         if stage == 'test' or stage is None:
-            self.mnist_test = TrialMNIST(root=self.data_dir, train=False, num_samples=32, download=True)
+            self.mnist_test = TrialMNIST(root=self.data_dir, train=False, num_samples=64, download=True)
             self.dims = getattr(self, 'dims', self.mnist_test[0][0].shape)
 
         self.non_picklable = lambda x: x**2

diff --git a/tests/models/test_amp.py b/tests/models/test_amp.py
@@ -15,7 +15,9 @@
 
 @pytest.mark.skipif(torch.cuda.device_count() < 2, reason="test requires multi-GPU machine")
 def test_multi_gpu_wandb_ddp_spawn(tmpdir):
-    """Make sure DP/DDP + AMP work."""
+    """
+    Test ddp + wb
+    """
     from pytorch_lightning.loggers import WandbLogger
     tutils.set_random_master_port()