mypy wandb ignore (axolotl-ai-cloud#572)

* mypy wandb ignore * fix isort for wandb
mkeoliya · Sep 14, 2023 · 51d3b69 · 51d3b69
1 parent 01d2e0e
commit 51d3b69
Showing 1 changed file with 2 additions and 2 deletions.
diff --git 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)