fix: sort values infer freq

Nixtla · Dec 9, 2023 · a611d73 · a611d73
1 parent 4b80613
commit a611d73
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/nbs/timegpt.ipynb b/nbs/timegpt.ipynb
@@ -276,7 +276,7 @@
     "        if self.freq is None or self.freq in special_freqs:\n",
     "            unique_id = df.iloc[0]['unique_id']\n",
     "            df_id = df.query('unique_id == @unique_id')\n",
-    "            inferred_freq = pd.infer_freq(df_id['ds'])\n",
+    "            inferred_freq = pd.infer_freq(df_id['ds'].sort_values())\n",
     "            if inferred_freq is None:\n",
     "                raise Exception(\n",
     "                    'Could not infer frequency of ds column. This could be due to '\n",

diff --git a/nixtlats/timegpt.py b/nixtlats/timegpt.py
@@ -218,7 +218,7 @@ def infer_freq(self, df: pd.DataFrame):
         if self.freq is None or self.freq in special_freqs:
             unique_id = df.iloc[0]["unique_id"]
             df_id = df.query("unique_id == @unique_id")
-            inferred_freq = pd.infer_freq(df_id["ds"])
+            inferred_freq = pd.infer_freq(df_id["ds"].sort_values())
             if inferred_freq is None:
                 raise Exception(
                     "Could not infer frequency of ds column. This could be due to "