feat: Add logger to write to csv files.

Signed-off-by: chaitanya-yeole <[email protected]>
Signed-off-by: ChaitanyaYeole02 <[email protected]>

Author: ChaitanyaYeole02

src/anomalib/loggers/__init__.py

@@ -17,13 +17,15 @@
     from .mlflow import AnomalibMLFlowLogger  # noqa: F401
     from .tensorboard import AnomalibTensorBoardLogger  # noqa: F401
     from .wandb import AnomalibWandbLogger  # noqa: F401
+    from .csv_logger import AnomalibCSVLogger
 
     __all__.extend(
         [
             "AnomalibCometLogger",
             "AnomalibTensorBoardLogger",
             "AnomalibWandbLogger",
             "AnomalibMLFlowLogger",
+            "AnomalibCSVLogger",
         ],
     )
 except ImportError:

src/anomalib/loggers/csv_logger.py

@@ -0,0 +1,19 @@
+from lightning.pytorch.loggers import CSVLogger
+
+
+class AnomalibCSVLogger(CSVLogger):
+    def __init__(
+            self,
+            save_dir: str,
+            name: str | None = "default",
+            version: int | str | None = None,
+            prefix: str = "",
+            flush_logs_every_n_step: int | None = 100,
+    ) -> None:
+        super().__init__(
+            save_dir=save_dir,
+            name=name,
+            version=version,
+            prefix=prefix,
+            flush_logs_every_n_steps=flush_logs_every_n_step,
+        )