From 83d81d9abe86e176f16be686b3a876d0624b7769 Mon Sep 17 00:00:00 2001 From: Alex Mallen Date: Thu, 21 Sep 2023 15:26:03 -0700 Subject: [PATCH] ignore type issue --- elk/extraction/extraction.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/elk/extraction/extraction.py b/elk/extraction/extraction.py index 7463b397..78bb5446 100644 --- a/elk/extraction/extraction.py +++ b/elk/extraction/extraction.py @@ -418,7 +418,7 @@ def select_hiddens( ) if cfg.get_lm_preds: out_record["lm_log_odds"] = torch.stack( - [d["lm_log_odds"] for d in ex] + [d["lm_log_odds"] for d in ex] # type: ignore ) del buffer[row_id] yield out_record