Upgrade wandb version

requirements/base.txt

@@ -8,6 +8,6 @@ pytorch-lightning>=1.6.0
 torchmetrics>=0.8.0
 torchvision>=0.9.1
 torchtext>=0.9.1
-wandb==0.12.9
+wandb==0.12.17
 matplotlib>=3.4.3
 gradio>=2.9.4

tests/pre_merge/utils/loggers/test_get_logger.py

@@ -14,6 +14,11 @@
 # See the License for the specific language governing permissions
 # and limitations under the License.
 
+from unittest.mock import patch
+
+patch("pytorch_lightning.utilities.imports._package_available", False)
+patch("pytorch_lightning.loggers.wandb.WandbLogger")
+
 import pytest
 from omegaconf import OmegaConf
 from pytorch_lightning.loggers import CSVLogger
@@ -37,36 +42,38 @@ def test_get_experiment_logger():
         }
     )
 
-    # get no logger
-    logger = get_experiment_logger(config=config)
-    assert isinstance(logger, bool)
-    config.project.logger = False
-    logger = get_experiment_logger(config=config)
-    assert isinstance(logger, bool)
+    with patch("pytorch_lightning.loggers.wandb.wandb"):
 
-    # get tensorboard
-    config.project.logger = "tensorboard"
-    logger = get_experiment_logger(config=config)
-    assert isinstance(logger[0], AnomalibTensorBoardLogger)
+        # get no logger
+        logger = get_experiment_logger(config=config)
+        assert isinstance(logger, bool)
+        config.project.logger = False
+        logger = get_experiment_logger(config=config)
+        assert isinstance(logger, bool)
 
-    # get wandb logger
-    config.project.logger = "wandb"
-    logger = get_experiment_logger(config=config)
-    assert isinstance(logger[0], AnomalibWandbLogger)
+        # get tensorboard
+        config.project.logger = "tensorboard"
+        logger = get_experiment_logger(config=config)
+        assert isinstance(logger[0], AnomalibTensorBoardLogger)
 
-    # get csv logger.
-    config.project.logger = "csv"
-    logger = get_experiment_logger(config=config)
-    assert isinstance(logger[0], CSVLogger)
+        # get wandb logger
+        config.project.logger = "wandb"
+        logger = get_experiment_logger(config=config)
+        assert isinstance(logger[0], AnomalibWandbLogger)
 
-    # get multiple loggers
-    config.project.logger = ["tensorboard", "wandb", "csv"]
-    logger = get_experiment_logger(config=config)
-    assert isinstance(logger[0], AnomalibTensorBoardLogger)
-    assert isinstance(logger[1], AnomalibWandbLogger)
-    assert isinstance(logger[2], CSVLogger)
+        # get csv logger.
+        config.project.logger = "csv"
+        logger = get_experiment_logger(config=config)
+        assert isinstance(logger[0], CSVLogger)
 
-    # raise unknown
-    with pytest.raises(UnknownLogger):
-        config.project.logger = "randomlogger"
+        # get multiple loggers
+        config.project.logger = ["tensorboard", "wandb", "csv"]
         logger = get_experiment_logger(config=config)
+        assert isinstance(logger[0], AnomalibTensorBoardLogger)
+        assert isinstance(logger[1], AnomalibWandbLogger)
+        assert isinstance(logger[2], CSVLogger)
+
+        # raise unknown
+        with pytest.raises(UnknownLogger):
+            config.project.logger = "randomlogger"
+            logger = get_experiment_logger(config=config)