From f36664ba5cb006bd0ec5999b0297975f34508f80 Mon Sep 17 00:00:00 2001 From: Wing Lian Date: Thu, 14 Sep 2023 11:17:30 -0400 Subject: [PATCH] mypy wandb ignore (#572) * mypy wandb ignore * fix isort for wandb --- src/axolotl/utils/callbacks.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) 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)