diff --git a/pandas_flavor/register.py b/pandas_flavor/register.py index 80c7fdb..d8c20e7 100644 --- a/pandas_flavor/register.py +++ b/pandas_flavor/register.py @@ -19,7 +19,7 @@ def print_column(df, col): def inner(*args, **kwargs): class AccessorMethod(object): - + def __init__(self, pandas_obj): self._obj = pandas_obj @@ -30,6 +30,8 @@ def __call__(self, *args, **kwargs): register_dataframe_accessor(method.__name__)(AccessorMethod) + return method + return inner() @@ -50,4 +52,6 @@ def __call__(self, *args, **kwargs): register_series_accessor(method.__name__)(AccessorMethod) + return method + return inner()