add wandb tags

Author: sfluegel

chebai/loggers/custom.py

@@ -1,5 +1,5 @@
 from datetime import datetime
-from typing import Literal, Optional, Union
+from typing import Literal, Optional, Union, List
 import os
 
 from lightning.fabric.utilities.types import _PATH
@@ -22,13 +22,16 @@ def __init__(
         entity: Optional[str] = None,
         offline: bool = False,
         log_model: Union[Literal["all"], bool] = False,
+        verbose_hyperparameters: bool = False,
+        tags: Optional[List[str]] = None,
         **kwargs,
     ):
         if version is None:
             version = f"{datetime.now():%y%m%d-%H%M}"
         self._version = version
         self._name = name
         self._fold = fold
+        self.verbose_hyperparameters = verbose_hyperparameters
         super().__init__(
             name=self.name,
             save_dir=save_dir,
@@ -40,6 +43,8 @@ def __init__(
             offline=offline,
             **kwargs,
         )
+        if tags:
+            self.experiment.tags += tuple(tags)
 
     @property
     def name(self) -> Optional[str]: