From 5a0c36d8b6458acdbaa68c17cf9d0ea9559278fc Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Ond=C5=99ej=20Sojka?=
 <8470346+tensojka@users.noreply.github.com>
Date: Thu, 7 Dec 2023 14:19:55 +0100
Subject: [PATCH] Fix link in docstring in wikitext function

---
 transformer_lm/datasets.py | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/transformer_lm/datasets.py b/transformer_lm/datasets.py
index b741c220..f0e8ff51 100644
--- a/transformer_lm/datasets.py
+++ b/transformer_lm/datasets.py
@@ -40,7 +40,9 @@ def _load(save_dir, filenames):
 def wikitext(dataset="2", save_dir="/tmp"):
     """
     Load the WikiText-* language modeling dataset:
-        https://paperswithcode.com/dataset/penn-treebank
+        https://paperswithcode.com/dataset/wikitext-2
+        https://paperswithcode.com/dataset/wikitext-103
+        
     """
     if dataset not in ("2", "103"):
         raise ValueError(f'Dataset must be either "2" or "103", got {dataset}')