PL full test (#4309)

pyg-team · Mar 21, 2022 · 66d1521 · 66d1521
1 parent dec8dc0
commit 66d1521
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/test/data/test_lightning_datamodule.py b/test/data/test_lightning_datamodule.py
@@ -7,6 +7,7 @@
 
 from torch_geometric.data import LightningDataset, LightningNodeData
 from torch_geometric.nn import global_mean_pool
+from torch_geometric.testing import onlyFullTest
 
 try:
     from pytorch_lightning import LightningModule
@@ -53,6 +54,7 @@ def configure_optimizers(self):
         return torch.optim.Adam(self.parameters(), lr=0.01)
 
 
+@onlyFullTest
 @pytest.mark.skipif(no_pytorch_lightning, reason='PL not available')
 @pytest.mark.skipif(not torch.cuda.is_available(), reason='CUDA not available')
 @pytest.mark.parametrize('strategy', [None, 'ddp_spawn'])
@@ -135,6 +137,7 @@ def configure_optimizers(self):
         return torch.optim.Adam(self.parameters(), lr=0.01)
 
 
+@onlyFullTest
 @pytest.mark.skipif(no_pytorch_lightning, reason='PL not available')
 @pytest.mark.skipif(not torch.cuda.is_available(), reason='CUDA not available')
 @pytest.mark.parametrize('loader', ['full', 'neighbor'])
@@ -226,6 +229,7 @@ def configure_optimizers(self):
         return torch.optim.Adam(self.parameters(), lr=0.01)
 
 
+@onlyFullTest
 @pytest.mark.skipif(no_pytorch_lightning, reason='PL not available')
 @pytest.mark.skipif(not torch.cuda.is_available(), reason='CUDA not available')
 def test_lightning_hetero_node_data(get_dataset):