Merge pull request scikit-learn#4861 from ltiao/patch-1

Update plot_ols.py (trivial)

Author: agramfort

examples/linear_model/plot_ols.py

@@ -32,12 +32,11 @@
 
 
 # Use only one feature
-diabetes_X = diabetes.data[:, np.newaxis]
-diabetes_X_temp = diabetes_X[:, :, 2]
+diabetes_X = diabetes.data[:, np.newaxis, 2]
 
 # Split the data into training/testing sets
-diabetes_X_train = diabetes_X_temp[:-20]
-diabetes_X_test = diabetes_X_temp[-20:]
+diabetes_X_train = diabetes_X[:-20]
+diabetes_X_test = diabetes_X[-20:]
 
 # Split the targets into training/testing sets
 diabetes_y_train = diabetes.target[:-20]