🐞 Fix linting issues (#535)

openvinotoolkit · Sep 5, 2022 · 8e465ee · 8e465ee
1 parent 3ae63a1
commit 8e465ee
Show file tree

Hide file tree

Showing 5 changed files with 20 additions and 21 deletions.
diff --git a/anomalib/utils/callbacks/graph.py b/anomalib/utils/callbacks/graph.py
@@ -7,7 +7,11 @@
 from pytorch_lightning import Callback, LightningModule, Trainer
 from pytorch_lightning.utilities.cli import CALLBACK_REGISTRY
 
-from anomalib.utils.loggers import AnomalibTensorBoardLogger, AnomalibWandbLogger, AnomalibCometLogger
+from anomalib.utils.loggers import (
+    AnomalibCometLogger,
+    AnomalibTensorBoardLogger,
+    AnomalibWandbLogger,
+)
 
 
 @CALLBACK_REGISTRY

diff --git a/anomalib/utils/loggers/__init__.py b/anomalib/utils/loggers/__init__.py
@@ -12,9 +12,9 @@
 from omegaconf.listconfig import ListConfig
 from pytorch_lightning.loggers import CSVLogger, LightningLoggerBase
 
+from .comet import AnomalibCometLogger
 from .tensorboard import AnomalibTensorBoardLogger
 from .wandb import AnomalibWandbLogger
-from .comet import AnomalibCometLogger
 
 __all__ = [
     "AnomalibCometLogger",
@@ -116,7 +116,7 @@ def get_experiment_logger(
                     save_dir=wandb_logdir,
                 )
             )
-        elif experiment_logger=="comet":
+        elif experiment_logger == "comet":
             comet_logdir = os.path.join(config.project.path, "logs")
             os.makedirs(comet_logdir, exist_ok=True)
             run_name = (
@@ -125,11 +125,7 @@ def get_experiment_logger(
                 else f"{config.dataset.category} {config.model.name}"
             )
             logger_list.append(
-                AnomalibCometLogger(
-                    project_name=config.dataset.name,
-                    experiment_name=run_name,
-                    save_dir=comet_logdir
-                )
+                AnomalibCometLogger(project_name=config.dataset.name, experiment_name=run_name, save_dir=comet_logdir)
             )
         elif experiment_logger == "csv":
             logger_list.append(CSVLogger(save_dir=os.path.join(config.project.path, "logs")))

diff --git a/anomalib/utils/loggers/comet.py b/anomalib/utils/loggers/comet.py
@@ -10,7 +10,6 @@
 from pytorch_lightning.loggers.comet import CometLogger
 from pytorch_lightning.utilities import rank_zero_only
 
-
 from .base import ImageLoggerBase
 
 

diff --git a/docs/blog/001-train-custom-dataset/README.md b/docs/blog/001-train-custom-dataset/README.md
@@ -181,16 +181,16 @@ To use TensorBoard and/or W&B logger and/or Comet logger, ensure that the logger
 An example configuration for saving to TensorBoard is shown in the figure below. Similarly after setting logger to `wandb` or 'comet' you will see the images on your wandb and/or comet project dashboard.
 
 ```yaml
-  visualization:
-    show_images: False # show images on the screen
-    save_images: False # save images to the file system
-    log_images: True # log images to the available loggers (if any)
-    image_save_path: null # path to which images will be saved
-    mode: full # options: ["full", "simple"]
-
-  logging:
-    logger: [comet, tensorboard, wandb] #Choose any combination of these 3
-    log_graph: false
+visualization:
+  show_images: False # show images on the screen
+  save_images: False # save images to the file system
+  log_images: True # log images to the available loggers (if any)
+  image_save_path: null # path to which images will be saved
+  mode: full # options: ["full", "simple"]
+
+logging:
+  logger: [comet, tensorboard, wandb] #Choose any combination of these 3
+  log_graph: false
 ```
 
 <div align="center">

diff --git a/tests/pre_merge/utils/loggers/test_get_logger.py b/tests/pre_merge/utils/loggers/test_get_logger.py
@@ -13,9 +13,9 @@
 from pytorch_lightning.loggers import CSVLogger
 
 from anomalib.utils.loggers import (
+    AnomalibCometLogger,
     AnomalibTensorBoardLogger,
     AnomalibWandbLogger,
-    AnomalibCometLogger,
     UnknownLogger,
     get_experiment_logger,
 )
@@ -51,7 +51,7 @@ def test_get_experiment_logger():
         logger = get_experiment_logger(config=config)
         assert isinstance(logger[0], AnomalibWandbLogger)
 
-         # get comet logger
+        # get comet logger
         config.project.logger = "comet"
         logger = get_experiment_logger(config=config)
         assert isinstance(logger[0], AnomalibCometLogger)