From 122353c392b1a829ece179f7518ad3322255fe3e Mon Sep 17 00:00:00 2001 From: Simon Mo Date: Wed, 13 May 2020 14:09:57 -0700 Subject: [PATCH] [Serve] Fix SKLearn example against newest version (#8428) --- python/ray/serve/examples/doc/tutorial_sklearn.py | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/python/ray/serve/examples/doc/tutorial_sklearn.py b/python/ray/serve/examples/doc/tutorial_sklearn.py index 075062d4a..57c812cdd 100644 --- a/python/ray/serve/examples/doc/tutorial_sklearn.py +++ b/python/ray/serve/examples/doc/tutorial_sklearn.py @@ -15,8 +15,9 @@ from sklearn.metrics import mean_squared_error # __doc_train_model_begin__ # Load data -data, target, target_names, description, feature_names, _ = load_iris().values( -) +iris_dataset = load_iris() +data, target, target_names = iris_dataset["data"], iris_dataset[ + "target"], iris_dataset["target_names"] # Instantiate model model = GradientBoostingClassifier()