diff --git a/src/axolotl/utils/callbacks.py b/src/axolotl/utils/callbacks.py
index 819360f1d3..bf8c4145bd 100644
--- a/src/axolotl/utils/callbacks.py
+++ b/src/axolotl/utils/callbacks.py
@@ -385,7 +385,7 @@ def find_ranges(lst):
                 return ranges
 
             def log_table_from_dataloader(name: str, table_dataloader):
-                table = wandb.Table(
+                table = wandb.Table(  # type: ignore[attr-defined]
                     columns=[
                         "id",
                         "Prompt",
@@ -506,7 +506,7 @@ def log_table_from_dataloader(name: str, table_dataloader):
                         )
                         row_index += 1
 
-                wandb.run.log({f"{name} - Predictions vs Ground Truth": table})
+                wandb.run.log({f"{name} - Predictions vs Ground Truth": table})  # type: ignore[attr-defined]
 
             if is_main_process():
                 log_table_from_dataloader("Eval", eval_dataloader)