Skip to content

Commit 95e5718

Browse files
adarobcopybara-github
authored andcommitted
Handle cases when TriviaQA examples are missing the SearchResults or EntityPages field.
PiperOrigin-RevId: 283911282
1 parent 11503d1 commit 95e5718

File tree

1 file changed

+4
-2
lines changed

1 file changed

+4
-2
lines changed

tensorflow_datasets/text/trivia_qa.py

Lines changed: 4 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -307,12 +307,14 @@ def _transpose_and_strip_dicts(dicts, field_names):
307307
}
308308

309309
search_results = _transpose_and_strip_dicts(
310-
_add_context(article["SearchResults"], "SearchContext", web_dir),
310+
_add_context(
311+
article.get("SearchResults", []), "SearchContext", web_dir),
311312
["Description", "Filename", "Rank", "Title", "Url",
312313
"SearchContext"])
313314

314315
entity_pages = _transpose_and_strip_dicts(
315-
_add_context(article["EntityPages"], "WikiContext", wiki_dir),
316+
_add_context(
317+
article.get("EntityPages", []), "WikiContext", wiki_dir),
316318
["DocSource", "Filename", "Title", "WikiContext"])
317319

318320
question = article["Question"].strip()

0 commit comments

Comments
 (0)