Add "return method"

pyjanitor-devs · Jul 18, 2018 · 02e849c · 02e849c
1 parent bb89234
commit 02e849c
Showing 1 changed file with 5 additions and 1 deletion.
diff --git 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()