minor

apache · Jun 18, 2015 · 81482fd · 81482fd
1 parent 5d9fe61
commit 81482fd
Showing 1 changed file with 2 additions and 10 deletions.
diff --git a/python/pyspark/mllib/clustering.py b/python/pyspark/mllib/clustering.py
@@ -445,23 +445,15 @@ def predictOn(self, dstream):
         Returns a transformed dstream object
         """
         self._validate(dstream)
-
-        def predict(ds):
-            return self.latestModel.predict(ds)
-
-        return dstream.map(predict)
+        return dstream.map(lambda x: self.latestModel.predict(x))
 
     def predictOnValues(self, dstream):
         """
         Make predictions on a keyed dstream.
         Returns a transformed dstream object.
         """
         self._validate(dstream)
-
-        def predict(ds):
-            return self.latestModel.predict(ds)
-
-        return dstream.mapValues(predict)
+        return dstream.mapValues(lambda x: self.latestModel.predict(x))
 
 
 def _test():