cli: allow specifying summary epochs to calculate

locklost/summary.py

@@ -359,6 +359,9 @@ def _parser_add_arguments(parser):
     parser.add_argument(
         'path',
         help="summary plots directory path")
+    parser.add_argument(
+        'epoch', nargs='*', default=list(EPOCHS),
+        help="epoch to calculate {}, default all".format(list(EPOCHS)))
 
 
 def main(args=None):
@@ -368,12 +371,20 @@ def main(args=None):
         _parser_add_arguments(parser)
         args = parser.parse_args()
 
+    for epoch in args.epoch:
+        if epoch not in set(EPOCHS):
+            msg = "'{}' is not a valid epoch.".format(epoch)
+            try:
+                parser.error(msg)
+            except UnboundLocalError:
+                raise SystemExit(msg)
+
     try:
         os.mkdir(args.path)
     except FileExistsError:
         pass
 
-    for epoch in EPOCHS:
+    for epoch in args.epoch:
         print('summarizing locklosses epoch {}'.format(epoch))
         plot_summary(args.path, epoch)