Stop showing any tracebacks in the console

labelmeai · Jan 21, 2025 · ede350f · ede350f
1 parent f0cb08f
commit ede350f
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/labelme_toolkit/_cli/__init__.py b/labelme_toolkit/_cli/__init__.py
@@ -17,11 +17,16 @@
 @click.version_option(__version__)
 def cli():
     logger.remove(0)
+
+    def format_console(record: dict):
+        # don't show any tracebacks in the console
+        return "<level>{message}</level>\n"
+
     logger.add(
         sys.stderr,
         level="INFO",
         colorize=True,
-        format="<level>{message}</level>",
+        format=format_console,
         backtrace=False,
         diagnose=False,
     )