remove = for float

Author: Davies Liu

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