aboutsummaryrefslogtreecommitdiff
path: root/python/pyspark/storagelevel.py
diff options
context:
space:
mode:
authorprabinb <prabin.banka@imaginea.com>2014-03-11 23:57:05 -0700
committerPatrick Wendell <pwendell@gmail.com>2014-03-11 23:57:05 -0700
commitaf7f2f10902c7b42e08797f7467dd06e4803594c (patch)
tree595b6989fc69b900d2d4f120d915d6f8a453c7d3 /python/pyspark/storagelevel.py
parent2409af9dcf238e1ad87080a389e05a696c41dc72 (diff)
downloadspark-af7f2f10902c7b42e08797f7467dd06e4803594c.tar.gz
spark-af7f2f10902c7b42e08797f7467dd06e4803594c.tar.bz2
spark-af7f2f10902c7b42e08797f7467dd06e4803594c.zip
Spark-1163, Added missing Python RDD functions
Author: prabinb <prabin.banka@imaginea.com> Closes #92 from prabinb/python-api-rdd and squashes the following commits: 51129ca [prabinb] Added missing Python RDD functions Added __repr__ function to StorageLevel class. Added doctest for RDD.getStorageLevel().
Diffstat (limited to 'python/pyspark/storagelevel.py')
-rw-r--r--python/pyspark/storagelevel.py4
1 files changed, 4 insertions, 0 deletions
diff --git a/python/pyspark/storagelevel.py b/python/pyspark/storagelevel.py
index b31f4762e6..c3e3a44e8e 100644
--- a/python/pyspark/storagelevel.py
+++ b/python/pyspark/storagelevel.py
@@ -31,6 +31,10 @@ class StorageLevel:
self.deserialized = deserialized
self.replication = replication
+ def __repr__(self):
+ return "StorageLevel(%s, %s, %s, %s)" % (
+ self.useDisk, self.useMemory, self.deserialized, self.replication)
+
StorageLevel.DISK_ONLY = StorageLevel(True, False, False)
StorageLevel.DISK_ONLY_2 = StorageLevel(True, False, False, 2)
StorageLevel.MEMORY_ONLY = StorageLevel(False, True, True)