Get package choices from source of truth

bpfarmer · Jul 11, 2016 · 8d976fe · 8d976fe
1 parent c3b03b8
commit 8d976fe
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/generate-flag-types b/generate-flag-types
@@ -78,7 +78,7 @@ def main(sysargs=sys.argv[:]):
         formatter_class=_FancyFormatter)
     parser.add_argument(
         'package',
-        type=str, default='cli', choices=['cli', 'altsrc'],
+        type=str, default='cli', choices=_WRITEFUNCS.keys(),
         help='Package for which flag types will be generated'
     )
     parser.add_argument(