1 files changed, 11 insertions, 1 deletions
diff --git a/scripts/cli/options.py b/scripts/cli/options.py
index 67f35dc..58d3df4 100644
--- a/scripts/cli/options.py
+++ b/scripts/cli/options.py
@@ -13,7 +13,7 @@ def add_logging(parser: ArgumentParser) -> ArgumentParser:
         type=str,
         default="INFO",
         choices=loglevels,
-        help=(f"Controls the severity of events to log. Valid values are: " +
+        help=("Controls the severity of events to log. Valid values are: " +
               ", ".join(f"'{level}'" for level in loglevels)))
     return parser
 
@@ -44,3 +44,13 @@ def add_population_id(parser: ArgumentParser) -> ArgumentParser:
                         type=int,
                         help="The ID for the population to operate on.")
     return parser
+
+
+def add_dataset_id(parser: ArgumentParser) -> ArgumentParser:
+    """Add dataset-id as a mandatory argument."""
+    parser = add_population_id(parser)
+    parser.add_argument("dataset_id",
+                        metavar="DATASET-ID",
+                        type=int,
+                        help="The ID for the dataset to operate on.")
+    return parser