aboutsummaryrefslogtreecommitdiffstats
path: root/utils
diff options
context:
space:
mode:
Diffstat (limited to 'utils')
-rw-r--r--utils/misc.py14
1 files changed, 12 insertions, 2 deletions
diff --git a/utils/misc.py b/utils/misc.py
index c54d25c..2ff0664 100644
--- a/utils/misc.py
+++ b/utils/misc.py
@@ -11,9 +11,12 @@ from __future__ import absolute_import, division
import os
import errno
-from collections import Sequence
import multiprocessing
+import argparse
+from collections import Sequence
+from operator import attrgetter
+
import numpy as np
from utils.enums import Likelihood, ParamTag
@@ -152,6 +155,13 @@ class ParamSet(Sequence):
if obj.tag is tag])
+class SortingHelpFormatter(argparse.HelpFormatter):
+ """Sort argparse help options alphabetically."""
+ def add_arguments(self, actions):
+ actions = sorted(actions, key=attrgetter('option_strings'))
+ super(SortingHelpFormatter, self).add_arguments(actions)
+
+
def gen_outfile_name(args):
"""Generate a name for the output file based on the input args.
@@ -238,7 +248,7 @@ def print_args(args):
"""
arg_vars = vars(args)
- for key in arg_vars.iterkeys():
+ for key in sorted(arg_vars):
print '== {0:<25} = {1}'.format(key, arg_vars[key])