Skip to content

Commit 8806d9d

Browse files
committed
fix import
1 parent 4e24834 commit 8806d9d

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

tests/deprecated_api/test_remove_1-7.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -18,7 +18,8 @@
1818
import torch
1919

2020
from pytorch_lightning import Callback, LightningDataModule, Trainer
21-
from pytorch_lightning.callbacks.xla_stats_monitor import GPUStatsMonitor, XLAStatsMonitor
21+
from pytorch_lightning.callbacks.gpu_stats_monitor import GPUStatsMonitor
22+
from pytorch_lightning.callbacks.xla_stats_monitor import XLAStatsMonitor
2223
from pytorch_lightning.loggers import LoggerCollection, TestTubeLogger
2324
from tests.deprecated_api import _soft_unimport_module
2425
from tests.helpers import BoringModel

0 commit comments

Comments
 (0)