Merge pull request #22 from yumemio/main

fix: `NotImplementedError` in download_dataset
AdityaNG · Jul 14, 2024 · 31ad768 · 31ad768
2 parents 7117646 + 6a3abd8
commit 31ad768
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/kan_gpt/download_dataset.py b/kan_gpt/download_dataset.py
@@ -44,8 +44,8 @@ def main(args):
         download_webtext()
     elif args.dataset == "tinyshakespeare":
         download_tinyshakespeare()
-
-    raise NotImplementedError
+    else:
+        raise NotImplementedError
 
 
 if __name__ == "__main__":