change column names

Nixtla · Nov 27, 2024 · e3fa7ce · e3fa7ce
1 parent a0f3dfa
commit e3fa7ce
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 57 deletions.
diff --git a/nbs/src/nixtla_client.ipynb b/nbs/src/nixtla_client.ipynb
@@ -154,18 +154,7 @@
    "cell_type": "code",
    "execution_count": null,
    "metadata": {},
-   "outputs": [
-    {
-     "data": {
-      "text/plain": [
-       "True"
-      ]
-     },
-     "execution_count": null,
-     "metadata": {},
-     "output_type": "execute_result"
-    }
-   ],
+   "outputs": [],
    "source": [
     "#| hide\n",
     "load_dotenv(override=True)"
@@ -1815,10 +1804,9 @@
     "            }\n",
     "        )\n",
     "        out = ufp.assign_columns(out, 'TimeGPT', resp['mean'])\n",
-    "        out_aggregated = (out.groupby(['unique_id', 'ds'])\n",
-    "                   .agg({'TimeGPT': 'median', 'y': 'first'})\n",
-    "                   .reset_index())\n",
-    "        out = (out_aggregated.groupby('unique_id')\n",
+    "        out_aggregated = (out.groupby([id_col, time_col], as_index=False)\n",
+    "                   .agg({'TimeGPT': 'median', target_col: 'first'}))\n",
+    "        out = (out_aggregated.groupby(id_col)\n",
     "                .tail(detection_size)\n",
     "                .reset_index(drop=True))\n",
     "        out = ufp.assign_columns(out, 'anomaly', resp['anomaly'])\n",
@@ -2549,18 +2537,7 @@
    "cell_type": "code",
    "execution_count": null,
    "metadata": {},
-   "outputs": [
-    {
-     "data": {
-      "text/plain": [
-       "True"
-      ]
-     },
-     "execution_count": null,
-     "metadata": {},
-     "output_type": "execute_result"
-    }
-   ],
+   "outputs": [],
    "source": [
     "#| hide\n",
     "nixtla_client.validate_api_key()"
@@ -3544,26 +3521,7 @@
    "cell_type": "code",
    "execution_count": null,
    "metadata": {},
-   "outputs": [
-    {
-     "name": "stderr",
-     "output_type": "stream",
-     "text": [
-      "INFO:__main__:Validating inputs...\n",
-      "INFO:__main__:Preprocessing dataframes...\n",
-      "INFO:__main__:Calling Online Anomaly Detector Endpoint...\n"
-     ]
-    },
-    {
-     "name": "stderr",
-     "output_type": "stream",
-     "text": [
-      "INFO:__main__:Validating inputs...\n",
-      "INFO:__main__:Preprocessing dataframes...\n",
-      "INFO:__main__:Calling Online Anomaly Detector Endpoint...\n"
-     ]
-    }
-   ],
+   "outputs": [],
    "source": [
     "#| hide\n",
     "# Test real-time anomaly detection\n",

diff --git a/nixtla/nixtla_client.py b/nixtla/nixtla_client.py
@@ -1746,16 +1746,10 @@ def detect_anomalies_realtime(
             }
         )
         out = ufp.assign_columns(out, "TimeGPT", resp["mean"])
-        out_aggregated = (
-            out.groupby(["unique_id", "ds"])
-            .agg({"TimeGPT": "median", "y": "first"})
-            .reset_index()
-        )
-        out = (
-            out_aggregated.groupby("unique_id")
-            .tail(detection_size)
-            .reset_index(drop=True)
+        out_aggregated = out.groupby([id_col, time_col], as_index=False).agg(
+            {"TimeGPT": "median", target_col: "first"}
         )
+        out = out_aggregated.groupby(id_col).tail(detection_size).reset_index(drop=True)
         out = ufp.assign_columns(out, "anomaly", resp["anomaly"])
         out = ufp.assign_columns(out, "anomaly_score", resp["anomaly_score"])
         return _maybe_add_intervals(out, resp["intervals"])