Merge pull request #4 from ericmjl/return_fix

Add "return method"

Author: Zsailer

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()