aboutsummaryrefslogtreecommitdiff
path: root/python/pyspark/mllib/linalg.py
diff options
context:
space:
mode:
authorDoris Xin <doris.s.xin@gmail.com>2014-07-31 20:32:57 -0700
committerXiangrui Meng <meng@databricks.com>2014-07-31 20:32:57 -0700
commitd8430148ee1f6ba02569db0538eeae473a32c78e (patch)
treed5103a5bc8f3068c48e0d581abe515560c1ecfe5 /python/pyspark/mllib/linalg.py
parent8f51491ea78d8e88fc664c2eac3b4ac14226d98f (diff)
downloadspark-d8430148ee1f6ba02569db0538eeae473a32c78e.tar.gz
spark-d8430148ee1f6ba02569db0538eeae473a32c78e.tar.bz2
spark-d8430148ee1f6ba02569db0538eeae473a32c78e.zip
[SPARK-2724] Python version of RandomRDDGenerators
RandomRDDGenerators but without support for randomRDD and randomVectorRDD, which take in arbitrary DistributionGenerator. `randomRDD.py` is named to avoid collision with the built-in Python `random` package. Author: Doris Xin <doris.s.xin@gmail.com> Closes #1628 from dorx/pythonRDD and squashes the following commits: 55c6de8 [Doris Xin] review comments. all python units passed. f831d9b [Doris Xin] moved default args logic into PythonMLLibAPI 2d73917 [Doris Xin] fix for linalg.py 8663e6a [Doris Xin] reverting back to a single python file for random f47c481 [Doris Xin] docs update 687aac0 [Doris Xin] add RandomRDDGenerators.py to run-tests 4338f40 [Doris Xin] renamed randomRDD to rand and import as random 29d205e [Doris Xin] created mllib.random package bd2df13 [Doris Xin] typos 07ddff2 [Doris Xin] units passed. 23b2ecd [Doris Xin] WIP
Diffstat (limited to 'python/pyspark/mllib/linalg.py')
-rw-r--r--python/pyspark/mllib/linalg.py4
1 files changed, 4 insertions, 0 deletions
diff --git a/python/pyspark/mllib/linalg.py b/python/pyspark/mllib/linalg.py
index 71f4ad1a8d..54720c2324 100644
--- a/python/pyspark/mllib/linalg.py
+++ b/python/pyspark/mllib/linalg.py
@@ -255,4 +255,8 @@ def _test():
exit(-1)
if __name__ == "__main__":
+ # remove current path from list of search paths to avoid importing mllib.random
+ # for C{import random}, which is done in an external dependency of pyspark during doctests.
+ import sys
+ sys.path.pop(0)
_test()