From 930b81f96cb93d3a2e4e3c3b5f0c019ce648a6d1 Mon Sep 17 00:00:00 2001 From: ananthsub Date: Sun, 22 Aug 2021 04:58:48 -0700 Subject: [PATCH] Remove unused rank_zero_deprecation in WandB logger (#9034) * Remove unused imports in WandB logger and corresponding test --- pytorch_lightning/loggers/wandb.py | 2 +- tests/loggers/test_wandb.py | 1 - 2 files changed, 1 insertion(+), 2 deletions(-) diff --git a/pytorch_lightning/loggers/wandb.py b/pytorch_lightning/loggers/wandb.py index 94baf1781e..0299cb522e 100644 --- a/pytorch_lightning/loggers/wandb.py +++ b/pytorch_lightning/loggers/wandb.py @@ -29,7 +29,7 @@ from pytorch_lightning.loggers.base import LightningLoggerBase, rank_zero_experi from pytorch_lightning.utilities import _module_available, rank_zero_only from pytorch_lightning.utilities.exceptions import MisconfigurationException from pytorch_lightning.utilities.imports import _compare_version -from pytorch_lightning.utilities.warnings import rank_zero_deprecation, rank_zero_warn +from pytorch_lightning.utilities.warnings import rank_zero_warn _WANDB_AVAILABLE = _module_available("wandb") _WANDB_GREATER_EQUAL_0_10_22 = _compare_version("wandb", operator.ge, "0.10.22") diff --git a/tests/loggers/test_wandb.py b/tests/loggers/test_wandb.py index e5b80993b3..0684727e84 100644 --- a/tests/loggers/test_wandb.py +++ b/tests/loggers/test_wandb.py @@ -18,7 +18,6 @@ from unittest import mock import pytest -import pytorch_lightning from pytorch_lightning import Trainer from pytorch_lightning.loggers import WandbLogger from pytorch_lightning.utilities.exceptions import MisconfigurationException