aboutsummaryrefslogtreecommitdiff
path: root/python/pyspark/mllib/regression.py
diff options
context:
space:
mode:
authorElisey Zanko <elisey.zanko@gmail.com>2015-04-20 10:44:09 -0700
committerJosh Rosen <joshrosen@databricks.com>2015-04-20 10:44:09 -0700
commit77176619a97d07811ab20e1dde4677359d85eb33 (patch)
tree3d362532bd4a21d2c888425791418e6001485d80 /python/pyspark/mllib/regression.py
parent968ad972175390bb0a96918fd3c7b318d70fa466 (diff)
downloadspark-77176619a97d07811ab20e1dde4677359d85eb33.tar.gz
spark-77176619a97d07811ab20e1dde4677359d85eb33.tar.bz2
spark-77176619a97d07811ab20e1dde4677359d85eb33.zip
[SPARK-6661] Python type errors should print type, not object
Author: Elisey Zanko <elisey.zanko@gmail.com> Closes #5361 from 31z4/spark-6661 and squashes the following commits: 73c5d79 [Elisey Zanko] Python type errors should print type, not object
Diffstat (limited to 'python/pyspark/mllib/regression.py')
-rw-r--r--python/pyspark/mllib/regression.py2
1 files changed, 1 insertions, 1 deletions
diff --git a/python/pyspark/mllib/regression.py b/python/pyspark/mllib/regression.py
index cd7310a64f..a0117c5713 100644
--- a/python/pyspark/mllib/regression.py
+++ b/python/pyspark/mllib/regression.py
@@ -170,7 +170,7 @@ def _regression_train_wrapper(train_func, modelClass, data, initial_weights):
from pyspark.mllib.classification import LogisticRegressionModel
first = data.first()
if not isinstance(first, LabeledPoint):
- raise ValueError("data should be an RDD of LabeledPoint, but got %s" % first)
+ raise TypeError("data should be an RDD of LabeledPoint, but got %s" % type(first))
if initial_weights is None:
initial_weights = [0.0] * len(data.first().features)
if (modelClass == LogisticRegressionModel):