From f85dc71f46ae0179fd78377e358dae77d0ac33e9 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Ph=C3=BAc=20H=2E=20L=C3=AA=20Kh=E1=BA=AFc?=
 <phuc@lkhphuc.com>
Date: Wed, 2 Aug 2023 16:23:57 +0200
Subject: [PATCH] Update pytorch_lightning.py

update deprecated property
---
 nvitop/callbacks/pytorch_lightning.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/nvitop/callbacks/pytorch_lightning.py b/nvitop/callbacks/pytorch_lightning.py
index 0c49ce4..1bd68a9 100644
--- a/nvitop/callbacks/pytorch_lightning.py
+++ b/nvitop/callbacks/pytorch_lightning.py
@@ -119,7 +119,7 @@ class GpuStatsLogger(Callback):  # pylint: disable=too-many-instance-attributes
                 f'The root device type is {trainer.strategy.root_device.type}.',
             )
 
-        device_ids = trainer.data_parallel_device_ids
+        device_ids = trainer.device_ids
         try:
             self._devices = get_devices_by_logical_ids(device_ids, unique=True)
         except (libnvml.NVMLError, RuntimeError) as ex: