test

Lightning-AI · Jul 28, 2021 · fb1051b · fb1051b
1 parent ded1b12
commit fb1051b
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/tests/trainer/logging_/test_train_loop_logging.py b/tests/trainer/logging_/test_train_loop_logging.py
@@ -24,7 +24,6 @@
 import torch
 from torchmetrics import Accuracy
 
-import pytorch_lightning as pl
 from pytorch_lightning import callbacks, Trainer
 from pytorch_lightning.callbacks import EarlyStopping, ModelCheckpoint
 from pytorch_lightning.utilities.exceptions import MisconfigurationException
@@ -581,7 +580,7 @@ def test_metric_are_properly_reduced(tmpdir):
     class TestingModel(BoringModel):
         def __init__(self, *args, **kwargs) -> None:
             super().__init__()
-            self.val_acc = pl.metrics.Accuracy()
+            self.val_acc = Accuracy()
 
         def training_step(self, batch, batch_idx):
             output = super().training_step(batch, batch_idx)