diff options
author | Holden Karau <holden@us.ibm.com> | 2016-12-20 15:51:21 -0800 |
---|---|---|
committer | Reynold Xin <rxin@databricks.com> | 2016-12-20 15:51:21 -0800 |
commit | 047a9d92caa1f0af2305e4afeba8339abf32518b (patch) | |
tree | ae8588410abee1171dc0cf18ffcbd71a99f62399 /python/pyspark/__init__.py | |
parent | caed89321fdabe83e46451ca4e968f86481ad500 (diff) | |
download | spark-047a9d92caa1f0af2305e4afeba8339abf32518b.tar.gz spark-047a9d92caa1f0af2305e4afeba8339abf32518b.tar.bz2 spark-047a9d92caa1f0af2305e4afeba8339abf32518b.zip |
[SPARK-18576][PYTHON] Add basic TaskContext information to PySpark
## What changes were proposed in this pull request?
Adds basic TaskContext information to PySpark.
## How was this patch tested?
New unit tests to `tests.py` & existing unit tests.
Author: Holden Karau <holden@us.ibm.com>
Closes #16211 from holdenk/SPARK-18576-pyspark-taskcontext.
Diffstat (limited to 'python/pyspark/__init__.py')
-rw-r--r-- | python/pyspark/__init__.py | 5 |
1 files changed, 4 insertions, 1 deletions
diff --git a/python/pyspark/__init__.py b/python/pyspark/__init__.py index 5f93586a48..9331e74eed 100644 --- a/python/pyspark/__init__.py +++ b/python/pyspark/__init__.py @@ -34,6 +34,8 @@ Public classes: Access files shipped with jobs. - :class:`StorageLevel`: Finer-grained cache persistence levels. + - :class:`TaskContext`: + Information about the current running task, avaialble on the workers and experimental. """ @@ -49,6 +51,7 @@ from pyspark.accumulators import Accumulator, AccumulatorParam from pyspark.broadcast import Broadcast from pyspark.serializers import MarshalSerializer, PickleSerializer from pyspark.status import * +from pyspark.taskcontext import TaskContext from pyspark.profiler import Profiler, BasicProfiler from pyspark.version import __version__ @@ -106,5 +109,5 @@ from pyspark.sql import SQLContext, HiveContext, Row __all__ = [ "SparkConf", "SparkContext", "SparkFiles", "RDD", "StorageLevel", "Broadcast", "Accumulator", "AccumulatorParam", "MarshalSerializer", "PickleSerializer", - "StatusTracker", "SparkJobInfo", "SparkStageInfo", "Profiler", "BasicProfiler", + "StatusTracker", "SparkJobInfo", "SparkStageInfo", "Profiler", "BasicProfiler", "TaskContext", ] |