CLI: Require datasetid as an argument.

1 files changed, 4 insertions, 0 deletions

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())