From b727b223fb6d1e02a5eb4d7b1a6f5a79773fc648 Mon Sep 17 00:00:00 2001
From: Frederick Muriuki Muriithi
Date: Wed, 25 Mar 2026 13:03:22 -0500
Subject: Add the DATASET-ID argument to command-line interfaces.

---
 scripts/cli/options.py | 10 ++++++++++
 1 file changed, 10 insertions(+)

(limited to 'scripts/cli/options.py')

diff --git a/scripts/cli/options.py b/scripts/cli/options.py
index 70d2a27..58d3df4 100644
--- a/scripts/cli/options.py
+++ b/scripts/cli/options.py
@@ -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
-- 
cgit 1.4.1