WIP

1 files changed, 2 insertions, 2 deletions

diff --git a/make_dataset.py b/make_dataset.py
index ea02281..f996acc 100755
--- a/make_dataset.py
+++ b/make_dataset.py
@@ -33,7 +33,7 @@ def main():
     huggingface_hub.login()
     dataset = make_dataset()
 
-    splits = ds.train_test_split(
+    splits = dataset.train_test_split(
         test_size=0.1,
         seed=42,
     )
@@ -49,7 +49,7 @@ def main():
         "test": splits["test"],
     })
 
-    dataset.push_to_hub("asdf2k/caption_translation", repo_visibility="private")
+    dataset.push_to_hub("asdf2k/caption_translation", private=True)
 
 
 if __name__ == "__main__":