From 242491b97183203a77b5b8634f6e8825c8b101fa Mon Sep 17 00:00:00 2001
From: Frederick Muriuki Muriithi
Date: Thu, 11 Jan 2024 02:01:31 +0300
Subject: CLI: Require datasetid as an argument.

---
 scripts/rqtl2/install_genotypes.py | 4 ++++
 1 file changed, 4 insertions(+)

(limited to 'scripts/rqtl2')

diff --git a/scripts/rqtl2/install_genotypes.py b/scripts/rqtl2/install_genotypes.py
index 6ae3788..3b9e987 100644
--- a/scripts/rqtl2/install_genotypes.py
+++ b/scripts/rqtl2/install_genotypes.py
@@ -121,6 +121,7 @@ def cross_reference_genotypes(dbconn: mdb.Connection,
 def install_genotypes(dbconn: mdb.Connection,
                       speciesid: int,
                       populationid: int,
+                      datasetid: int,
                       rqtl2bundle: Path) -> int:
     """Load any existing genotypes into the database."""
     count = 0
@@ -190,6 +191,8 @@ if __name__ == "__main__":
                             help="Species to which bundle relates.")
         parser.add_argument("populationid",
                             help="Population to group data under")
+        parser.add_argument("datasetid",
+                            help="The dataset to which the data belongs.")
         parser.add_argument("rqtl2bundle",
                             help="Path to R/qtl2 bundle zip file.",
                             type=Path)
@@ -226,6 +229,7 @@ if __name__ == "__main__":
             return install_genotypes(dbconn,
                                      args.speciesid,
                                      args.populationid,
+                                     args.datasetid,
                                      args.rqtl2bundle)
 
     sys.exit(main())
-- 
cgit v1.2.3