diff options
Diffstat (limited to 'python/pyspark/rdd.py')
-rw-r--r-- | python/pyspark/rdd.py | 3 |
1 files changed, 2 insertions, 1 deletions
diff --git a/python/pyspark/rdd.py b/python/pyspark/rdd.py index 332258f5d1..58e1849cad 100644 --- a/python/pyspark/rdd.py +++ b/python/pyspark/rdd.py @@ -77,7 +77,8 @@ class RDD(object): have a storage level set yet. """ self.is_cached = True - self._jrdd.persist(storageLevel) + javaStorageLevel = self.ctx._getJavaStorageLevel(storageLevel) + self._jrdd.persist(javaStorageLevel) return self def unpersist(self): |