ignore type issue

EleutherAI · Sep 21, 2023 · 83d81d9 · 83d81d9
1 parent d48519d
commit 83d81d9
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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