Handle click groups in click helper

ProbablyFaiz · Jul 16, 2024 · 519b81a · 519b81a
1 parent 75db69c
commit 519b81a
Showing 1 changed file with 13 additions and 0 deletions.
diff --git a/rl/utils/click.py b/rl/utils/click.py
@@ -40,5 +40,18 @@ def decorator(f):
     return decorator
 
 
+def group(*args, **kwargs):
+    context_settings = kwargs.get("context_settings", {})
+    if "show_default" not in context_settings:
+        context_settings["show_default"] = True
+    kwargs["context_settings"] = context_settings
+
+    def decorator(f):
+        sys.excepthook = excepthook
+        return rich_click.group(*args, **kwargs)(f)
+
+    return decorator
+
+
 def __getattr__(name):
     return getattr(rich_click, name)