From 7f66bb029322a06b52359abf8836e158afaa6755 Mon Sep 17 00:00:00 2001
From: Frederick Muriuki Muriithi
Date: Mon, 20 Jun 2022 11:17:11 +0300
Subject: Restrict partial correlation method choices

- Have "Pearson's r" and "Spearman's rho" as the only valid choices for the
  partial correlations
---
 scripts/partial_correlations.py | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

(limited to 'scripts')

diff --git a/scripts/partial_correlations.py b/scripts/partial_correlations.py
index de364dc..d28a17a 100644
--- a/scripts/partial_correlations.py
+++ b/scripts/partial_correlations.py
@@ -106,7 +106,8 @@ def process_cli_arguments():
     parser.add_argument(
         "method",
         help="The correlation method to use",
-        type=str)
+        type=str,
+        choices=("Pearson's r", "Spearman's rho"))
     against_db_parser(against_traits_parser(
         parser.add_subparsers(
             title="subcommands",
-- 
cgit v1.2.3