Skip to content

Commit 97d465c

Browse files
committed
fix
1 parent 35aa966 commit 97d465c

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

python/pyspark/ml/regression.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -283,11 +283,11 @@ class LinearRegression(
283283
>>> model_path = temp_path + "/lr_model"
284284
>>> model.save(model_path)
285285
>>> model2 = LinearRegressionModel.load(model_path)
286-
>>> model.coefficients[0] == model2.coefficients[0]
286+
>>> bool(model.coefficients[0] == model2.coefficients[0])
287287
True
288-
>>> model.intercept == model2.intercept
288+
>>> bool(model.intercept == model2.intercept)
289289
True
290-
>>> model.transform(test0).take(1) == model2.transform(test0).take(1)
290+
>>> bool(model.transform(test0).take(1) == model2.transform(test0).take(1))
291291
True
292292
>>> model.numFeatures
293293
1

0 commit comments

Comments
 (0)