From 8aa01d2a6db0059077ecdd4e5948e6ffe51340ee Mon Sep 17 00:00:00 2001 From: Sasha Luccioni Date: Thu, 5 Aug 2021 02:56:13 -0400 Subject: [PATCH] Create perplexity.rst (#13004) Updating the import for load_dataset --- docs/source/perplexity.rst | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/docs/source/perplexity.rst b/docs/source/perplexity.rst index 2ad255e6d9..e3a84f7a25 100644 --- a/docs/source/perplexity.rst +++ b/docs/source/perplexity.rst @@ -96,7 +96,7 @@ dataset in memory. .. code-block:: python - from nlp import load_dataset + from datasets import load_dataset test = load_dataset('wikitext', 'wikitext-2-raw-v1', split='test') encodings = tokenizer('\n\n'.join(test['text']), return_tensors='pt')