RunIf(min_gpus=1)

Author: daniellepintz

pytorch_lightning/core/datamodule.py

@@ -22,7 +22,9 @@
 from pytorch_lightning.core.hooks import CheckpointHooks, DataHooks
 from pytorch_lightning.core.mixins import HyperparametersMixin
 from pytorch_lightning.utilities import rank_zero_deprecation
-from pytorch_lightning.utilities.argparse import add_argparse_args, from_argparse_args, get_init_arguments_and_types
+from pytorch_lightning.utilities.argparse import (add_argparse_args,
+                                                  from_argparse_args,
+                                                  get_init_arguments_and_types)
 
 
 class LightningDataModule(CheckpointHooks, DataHooks, HyperparametersMixin):

tests/deprecated_api/test_remove_1-7.py

@@ -370,12 +370,13 @@ def test_v1_7_0_weights_summary_trainer(tmpdir):
         t.weights_summary = "blah"
 
 
+@RunIf(min_gpus=1)
 def test_v1_7_0_deprecate_gpu_stats_monitor(tmpdir):
     with pytest.deprecated_call(match="The `GPUStatsMonitor` callback was deprecated in v1.5"):
         _ = GPUStatsMonitor()
 
 
 @RunIf(tpu=True)
-def test_v1_7_0_deprecate_tpu_stats_monitor(tmpdir):
+def test_v1_7_0_deprecate_xla_stats_monitor(tmpdir):
     with pytest.deprecated_call(match="The `XLAStatsMonitor` callback was deprecated in v1.5"):
         _ = XLAStatsMonitor()