code misspelling

sdpython · Sep 6, 2018 · 837b9f5 · 837b9f5
1 parent afc58f8
commit 837b9f5
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/_doc/notebooks/lectures/wines_color_linear.ipynb b/_doc/notebooks/lectures/wines_color_linear.ipynb
@@ -420,11 +420,11 @@
         "        pass\n",
         "    \n",
         "    def fit(X, y, cat):\n",
-        "        X = X.drop(cat, axis=1)\n",
+        "        X = X.drop(self.cat_col, axis=1)\n",
         "        # ...\n",
         "                \n",
         "    def predict(X, cat):\n",
-        "        X = X.drop(cat, axis=1)\n",
+        "        X = X.drop(self.cat_col, axis=1)\n",
         "        # ..."
       ]
     },