py27 fix

vaexio · Mar 18, 2019 · f8ec170 · f8ec170
1 parent 08b340d
commit f8ec170
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/packages/vaex-core/vaex/functions.py b/packages/vaex-core/vaex/functions.py
@@ -111,7 +111,10 @@ def f(function=function):
             def wrapper(*args, **kwargs):
                 return function(*args, **kwargs)
             return wrapper
-        function = functools.wraps(function)(f())
+        try:
+            function = functools.wraps(function)(f())
+        except AttributeError:
+            function = f()  # python 2 case
         register_function(name=name)(function)