fix Python UDF

apache · Sep 11, 2014 · 0a5b6eb · 0a5b6eb
1 parent 4b20494
commit 0a5b6eb
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/python/pyspark/sql.py b/python/pyspark/sql.py
@@ -971,7 +971,7 @@ def registerFunction(self, name, f, returnType=StringType()):
         [Row(c0=4)]
         """
         func = lambda _, it: imap(lambda x: f(*x), it)
-        command = (func,
+        command = (func, None,
                    BatchedSerializer(PickleSerializer(), 1024),
                    BatchedSerializer(PickleSerializer(), 1024))
         pickled_command = CloudPickleSerializer().dumps(command)