From b56d8677ad0ff8513e566334f4a78a24b88480c3 Mon Sep 17 00:00:00 2001
From: Jv Kyle Eclarin <mathemusician@users.noreply.github.com>
Date: Sat, 8 Jan 2022 10:24:29 -0600
Subject: [PATCH] Update test_pruning.py to use `devices` instead of `gpus` or
 `ipus` (#11339)

---
 tests/callbacks/test_device_stats_monitor.py | 9 ++++++---
 1 file changed, 6 insertions(+), 3 deletions(-)

diff --git a/tests/callbacks/test_device_stats_monitor.py b/tests/callbacks/test_device_stats_monitor.py
index e5b4516182..582a0471d7 100644
--- a/tests/callbacks/test_device_stats_monitor.py
+++ b/tests/callbacks/test_device_stats_monitor.py
@@ -44,7 +44,8 @@ def test_device_stats_gpu_from_torch(tmpdir):
         max_epochs=2,
         limit_train_batches=7,
         log_every_n_steps=1,
-        gpus=1,
+        accelerator="gpu",
+        devices=1,
         callbacks=[device_stats],
         logger=DebugLogger(tmpdir),
         enable_checkpointing=False,
@@ -73,7 +74,8 @@ def test_device_stats_gpu_from_nvidia(tmpdir):
         max_epochs=2,
         limit_train_batches=7,
         log_every_n_steps=1,
-        gpus=1,
+        accelerator="gpu",
+        devices=1,
         callbacks=[device_stats],
         logger=DebugLogger(tmpdir),
         enable_checkpointing=False,
@@ -101,7 +103,8 @@ def test_device_stats_monitor_tpu(tmpdir):
         default_root_dir=tmpdir,
         max_epochs=1,
         limit_train_batches=1,
-        tpu_cores=8,
+        accelerator="tpu",
+        devices=8,
         log_every_n_steps=1,
         callbacks=[device_stats],
         logger=DebugLogger(tmpdir),