tocmo0nlord/axolotl
1
0
Fork 0
Code Issues Pull Requests Actions 3 Packages Projects Releases Wiki Activity

mypy wandb ignore (#572)

Browse Source 
* mypy wandb ignore

* fix isort for wandb
This commit is contained in:
Wing Lian
2023-09-14 11:17:30 -04:00
committed by GitHub
parent 115795079d
commit c6d870b91d
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
src/axolotl/utils/callbacks.py
View File

@@ -385,7 +385,7 @@ def log_prediction_callback_factory(trainer: Trainer, tokenizer):
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_prediction_callback_factory(trainer: Trainer, tokenizer):
)
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)