Merge pull request #154 from cta-observatory/fix-make_progress_bar_op…

…tional Make progress bar optional and add it also to DL2 script
cta-observatory · Jul 20, 2021 · 52faa69 · 52faa69
2 parents 2fe78fc + de588cc
commit 52faa69
Show file tree

Hide file tree

Showing 2 changed files with 16 additions and 2 deletions.
diff --git a/protopipe/scripts/data_training.py b/protopipe/scripts/data_training.py
@@ -34,6 +34,12 @@ def main():
     parser.add_argument(
         "--debug", action="store_true", help="Print debugging information",
     )
+
+    parser.add_argument(
+        "--show_progress_bar",
+        action="store_true",
+        help="Show information about execution progress",
+    )
 
     parser.add_argument(
         "--save_images", action="store_true", help="Save also all images",
@@ -269,7 +275,8 @@ def main():
                                          debug=args.debug),
                     desc=source.__class__.__name__,
                     total=source.max_events,
-                    unit="event"
+                    unit="event",
+                    disable= not args.show_progress_bar
                  ):
 
             if good_event:

diff --git a/protopipe/scripts/write_dl2.py b/protopipe/scripts/write_dl2.py
@@ -35,6 +35,12 @@ def main():
     parser.add_argument(
         "--debug", action="store_true", help="Print debugging information",
     )
+
+    parser.add_argument(
+        "--show_progress_bar",
+        action="store_true",
+        help="Show information about execution progress",
+    )
 
     parser.add_argument("--regressor_dir", default="./", help="regressors directory")
     parser.add_argument("--classifier_dir", default="./", help="regressors directory")
@@ -314,7 +320,8 @@ class RecoEvent(tb.IsDescription):
                                          debug=args.debug),
                     desc=source.__class__.__name__,
                     total=source.max_events,
-                    unit="event"
+                    unit="event",
+                    disable= not args.show_progress_bar
                  ):
 
             # True direction