Update write_dl2.py

cta-observatory · Jul 15, 2021 · 21f3455 · 21f3455
1 parent 0676c72
commit 21f3455
Showing 1 changed file with 8 additions and 1 deletion.
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