use target_col in conformity scores (#134)

Nixtla · May 16, 2023 · 95129e3 · 95129e3
1 parent d4f16cc
commit 95129e3
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/mlforecast/forecast.py b/mlforecast/forecast.py
@@ -318,7 +318,7 @@ def _conformity_scores(
         for model in self.models.keys():
             # compute absolute error for each model
             cv_results[model] = np.abs(cv_results[model] - cv_results[target_col])
-        return cv_results.drop("y", axis=1)
+        return cv_results.drop(columns=target_col)
 
     def fit(
         self,

diff --git a/nbs/forecast.ipynb b/nbs/forecast.ipynb
@@ -416,7 +416,7 @@
     "        for model in self.models.keys():\n",
     "            # compute absolute error for each model\n",
     "            cv_results[model] = np.abs(cv_results[model] - cv_results[target_col])\n",
-    "        return cv_results.drop('y', axis=1)\n",
+    "        return cv_results.drop(columns=target_col)\n",
     "\n",
     "    def fit(\n",
     "        self,\n",