Merge pull request #49 from BrainLesion/48-readthedocs-infer-method

- fix citation decorator
BrainLesion · Apr 8, 2024 · 19c4895 · 19c4895
2 parents 1de48b7 + 9385e81
commit 19c4895
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/brainles_aurora/utils/console_decorators.py b/brainles_aurora/utils/console_decorators.py
@@ -1,9 +1,11 @@
+from functools import wraps
 from rich.console import Console
 
 CITATION_LINK = "https://github.com/BrainLesion/AURORA#citation"
 
 
 def citation_reminder(func):
+    @wraps(func)
     def wrapper(*args, **kwargs):
         console = Console()
         console.rule("Thank you for using [bold]AURORA[/bold]")