remove = for float

msjgriffiths · Nov 19, 2014 · ee17d78 · ee17d78
1 parent 13f7b05
commit ee17d78
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/python/pyspark/rddsampler.py b/python/pyspark/rddsampler.py
@@ -70,7 +70,7 @@ def func(self, split, iterator):
                     yield obj
         else:
             for obj in iterator:
-                if self.getUniformSample() <= self._fraction:
+                if self.getUniformSample() < self._fraction:
                     yield obj
 
 
@@ -106,5 +106,5 @@ def func(self, split, iterator):
                     yield key, val
         else:
             for key, val in iterator:
-                if self.getUniformSample() <= self._fractions[key]:
+                if self.getUniformSample() < self._fractions[key]:
                     yield key, val