fixed val-percentage and test-percentage being optional (#388)

v7labs · Apr 6, 2022 · e5f81bf · e5f81bf
1 parent d706872
commit e5f81bf
Showing 1 changed file with 2 additions and 4 deletions.
diff --git a/darwin/options.py b/darwin/options.py
@@ -190,10 +190,8 @@ def __init__(self):
             "split", help="Splits a local dataset following random and stratified split types."
         )
         parser_split.add_argument("dataset", type=str, help="Local dataset name to split.")
-        parser_split.add_argument("-v", "--val-percentage", type=float, required=True, help="Validation percentage.")
-        parser_split.add_argument(
-            "-t", "--test-percentage", type=float, required=False, default=0, help="Test percentage."
-        )
+        parser_split.add_argument("-v", "--val-percentage", required=True, type=float, help="Validation percentage.")
+        parser_split.add_argument("-t", "--test-percentage", required=True, type=float, help="Test percentage.")
         parser_split.add_argument("-s", "--seed", type=int, required=False, default=0, help="Split seed.")
 
         # List Files