fixing typing

chanind · May 6, 2022 · 6012b09 · 6012b09
1 parent 4bbee5c
commit 6012b09
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/frame_semantic_transformer/evaluate.py b/frame_semantic_transformer/evaluate.py
@@ -1,5 +1,5 @@
 from __future__ import annotations
-from typing import Iterable
+from typing import Sequence
 from tqdm import tqdm
 from transformers import T5ForConditionalGeneration, T5Tokenizer
 from nltk.corpus import framenet as fn
@@ -15,7 +15,7 @@
 def evaluate(
     model: T5ForConditionalGeneration,
     tokenizer: T5Tokenizer,
-    samples: Iterable[SampleSentence],
+    samples: Sequence[SampleSentence],
     batch_size: int = 10,
 ) -> dict[str, list[int]]:
     results: dict[str, list[int]] = {"frame": [0, 0, 0], "args": [0, 0, 0]}