fix function signature

pyjanitor-devs · Feb 16, 2018 · e4369ca · e4369ca
1 parent 245d4b0
commit e4369ca
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/pandas_flavor/register.py b/pandas_flavor/register.py
@@ -1,3 +1,4 @@
+from functools import wraps
 from .pandas_internals import (register_series_accessor,
                               register_dataframe_accessor)
 
@@ -18,11 +19,12 @@ def print_column(df, col):
     def inner(*args, **kwargs):
 
         class AccessorMethod(object):
-            __doc__ = method.__doc__
+
 
             def __init__(self, pandas_obj):
                 self._obj = pandas_obj
 
+            @wraps(method)
             def __call__(self, *args, **kwargs):
                 return method(self._obj, *args, **kwargs)
 
@@ -42,6 +44,7 @@ class AccessorMethod(object):
             def __init__(self, pandas_obj):
                 self._obj = pandas_obj
 
+            @wraps(method)
             def __call__(self, *args, **kwargs):
                 return method(self._obj, *args, **kwargs)