From 74738e414fcc3c85a4fa32aa2bd2bd1263405793 Mon Sep 17 00:00:00 2001
From: Xiaoyang Wu <xiaoyang.wu@connect.hku.hk>
Date: Thu, 5 Sep 2024 22:41:52 -0700
Subject: [PATCH] make --split optional, default download both split

---
 download_data.py | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/download_data.py b/download_data.py
index 3d8d5d1..27fedf9 100644
--- a/download_data.py
+++ b/download_data.py
@@ -216,7 +216,8 @@ def download_data(dataset,
 
     parser.add_argument(
         "--split",
-        choices=["Training", "Validation"],
+        choices=[None, "Training", "Validation"],
+        default=None
     )
 
     parser.add_argument(