@@ -85,7 +85,8 @@
     "X_train, X_test, y_train, y_test = train_test_split(\n",
     "    X, y, test_size=0.2, random_state=0)\n",
     "data = {\"train\": {\"X\": X_train, \"y\": y_train},\n",
-    "        \"test\": {\"X\": X_test, \"y\": y_test}}"
+    "        \"test\": {\"X\": X_test, \"y\": y_test}}\n",
+    "    return data"
    ]
   },
   {
Original file line number	Diff line number	Diff line change
`@@ -85,7 +85,8 @@`
`85`	`85`	`"X_train, X_test, y_train, y_test = train_test_split(\n",`
`86`	`86`	`" X, y, test_size=0.2, random_state=0)\n",`
`87`	`87`	`"data = {\"train\": {\"X\": X_train, \"y\": y_train},\n",`
`88`		`- " \"test\": {\"X\": X_test, \"y\": y_test}}"`
	`88`	`+ " \"test\": {\"X\": X_test, \"y\": y_test}}\n",`
	`89`	`+ " return data"`
`89`	`90`	`]`
`90`	`91`	`},`
`91`	`92`	`{`