diff options
Diffstat (limited to 'make_dataset.py')
| -rwxr-xr-x | make_dataset.py | 4 |
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__": |