preserve function signature and docstring in with_hv_extension (#601)

holoviz · Apr 27, 2021 · cccc1ce · cccc1ce
1 parent 73bbf1c
commit cccc1ce
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/hvplot/util.py b/hvplot/util.py
@@ -6,6 +6,7 @@
 import sys
 
 from distutils.version import LooseVersion
+from functools import wraps
 from types import FunctionType
 
 import pandas as pd
@@ -24,6 +25,7 @@
 
 def with_hv_extension(func, extension='bokeh', logo=False):
     """If hv.extension is not loaded, load before calling function"""
+    @wraps(func)
     def wrapper(*args, **kwargs):
         if extension and not getattr(hv.extension, '_loaded', False):
             hv.extension(extension, logo=logo)