Remove whitespace before parsing UIDs

pitt-crc · Dec 20, 2023 · 7d867be · 7d867be
1 parent c33a749
commit 7d867be
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/shinigami/cli.py b/shinigami/cli.py
@@ -58,13 +58,13 @@ def __init__(self) -> None:
         scan.set_defaults(callable=Application.scan)
         scan.add_argument('-c', '--clusters', nargs='+', required=True, help='cluster names to scan')
         scan.add_argument('-i', '--ignore-nodes', nargs='*', default=[], help='ignore the given nodes')
-        scan.add_argument('-u', '--uid-whitelist', nargs='+', type=loads, default=[0], help='user IDs to scan')
+        scan.add_argument('-u', '--uid-whitelist', nargs='+', type=lambda x: loads(x.replace(' ', '')), default=[0], help='user IDs to scan')
 
         # Subparser for the `Application.terminate` method
         terminate = subparsers.add_parser('terminate', parents=[common], help='terminate processes on a single node')
         terminate.set_defaults(callable=Application.terminate)
         terminate.add_argument('-n', '--nodes', nargs='+', required=True, help='the DNS name of the node to terminate')
-        terminate.add_argument('-u', '--uid-whitelist', nargs='+', type=loads, default=[0], help='user IDs to scan')
+        terminate.add_argument('-u', '--uid-whitelist', nargs='+', type=lambda x: loads(x.replace(' ', '')), default=[0], help='user IDs to scan')
 
 
 class Application: