diff --git a/sub-packages/bionemo-llm/src/bionemo/llm/utils/logger_utils.py b/sub-packages/bionemo-llm/src/bionemo/llm/utils/logger_utils.py
index 5f2f6b1957..ebba878c66 100644
--- a/sub-packages/bionemo-llm/src/bionemo/llm/utils/logger_utils.py
+++ b/sub-packages/bionemo-llm/src/bionemo/llm/utils/logger_utils.py
@@ -42,14 +42,14 @@ class WandbConfig(BaseModel):
         anonymous: Enables or explicitly disables anonymous logging.
     """  # noqa: D205
 
-    entity: str  # The team posting this run (default: your username or your default team)
+    entity: str | None  # The team posting this run (default: your username or your default team)
     project: str  # The name of the project to which this run will belong.
     # name: #Display name for the run. "This is handled by NeMoLogger"
     # save_dir: #Path where data is saved. "This is handled by NeMoLogger"
-    tags: List[str]  # Tags associated with this run.
-    group: str  # A unique string shared by all runs in a given group
+    tags: List[str] | None  # Tags associated with this run.
+    group: str | None  # A unique string shared by all runs in a given group
     offline: bool  # Run offline (data can be streamed later to wandb servers).
-    id: str  # Sets the version, mainly used to resume a previous run.
+    id: str | None  # Sets the version, mainly used to resume a previous run.
     anonymous: bool  # Enables or explicitly disables anonymous logging.
     log_model: bool  # Save checkpoints in wandb dir to upload on W&B servers.