diff --git a/python/pyspark/streaming/dstream.py b/python/pyspark/streaming/dstream.py index 4533c5d541a51..5d0dface2f043 100644 --- a/python/pyspark/streaming/dstream.py +++ b/python/pyspark/streaming/dstream.py @@ -150,6 +150,9 @@ def foreachRDD(self, func): """ Apply a function to each RDD in this DStream. """ + if func.func_code.co_argcount == 1: + old_func = func + func = lambda t, rdd: old_func(rdd) jfunc = TransformFunction(self._sc, func, self._jrdd_deserializer) api = self._ssc._jvm.PythonDStream api.callForeachRDD(self._jdstream, jfunc)