From 1af7511e5fbf3c6e89f0452623eaf663387ccfc2 Mon Sep 17 00:00:00 2001
From: pks <pks@pks.rocks>
Date: Sun, 30 Nov 2025 21:38:00 +0100
Subject: WIP

---
 make_dataset.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

(limited to 'make_dataset.py')

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__":
-- 
cgit v1.2.3