From 886301972ec1e7ed274040b46ec0e0c3f34c8ace Mon Sep 17 00:00:00 2001 From: Fan Ye Date: Fri, 21 Jun 2024 11:01:32 -0700 Subject: [PATCH] Fix import in TyDi model. PiperOrigin-RevId: 645441653 --- lit_nlp/examples/tydi/model.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/lit_nlp/examples/tydi/model.py b/lit_nlp/examples/tydi/model.py index a304dc5b..7f35f760 100644 --- a/lit_nlp/examples/tydi/model.py +++ b/lit_nlp/examples/tydi/model.py @@ -2,7 +2,7 @@ from collections.abc import Iterable from lit_nlp.api import model as lit_model from lit_nlp.api import types as lit_types -from lit_nlp.examples.datasets import question_answering +from lit_nlp.examples.tydi import data as tydi_data import numpy as np import transformers @@ -104,7 +104,7 @@ def input_spec(self): "question": lit_types.TextSegment(), "answers_text": lit_types.MultiSegmentAnnotations(), "language": lit_types.CategoryLabel( - required=False, vocab=question_answering.TYDI_LANG_VOCAB + required=False, vocab=tydi_data.TYDI_LANG_VOCAB ), }