From 82f10bd7944e7d5b093c946353d1a9e4da246a72 Mon Sep 17 00:00:00 2001 From: Ismael Juma Date: Wed, 1 Jun 2011 16:12:23 +0100 Subject: [PATCH] Remove unnecessary toStream calls. --- core/src/main/scala/spark/RDD.scala | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/core/src/main/scala/spark/RDD.scala b/core/src/main/scala/spark/RDD.scala index 590106388a..505ef76941 100644 --- a/core/src/main/scala/spark/RDD.scala +++ b/core/src/main/scala/spark/RDD.scala @@ -163,7 +163,7 @@ class FlatMappedRDD[U: ClassManifest, T: ClassManifest]( extends RDD[U](prev.context) { override def splits = prev.splits override val dependencies = List(new OneToOneDependency(prev)) - override def compute(split: Split) = prev.iterator(split).toStream.flatMap(f).iterator + override def compute(split: Split) = prev.iterator(split).flatMap(f) } class FilteredRDD[T: ClassManifest]( @@ -314,8 +314,8 @@ extends RDD[(K, U)](prev.context) { override val dependencies = List(new OneToOneDependency(prev)) override val partitioner = prev.partitioner override def compute(split: Split) = { - prev.iterator(split).toStream.flatMap { + prev.iterator(split).flatMap { case (k, v) => f(v).map(x => (k, x)) - }.iterator + } } }