Skip to content

Commit 40479a3

Browse files
committedNov 9, 2023
fix: add correct sort
1 parent 227cf96 commit 40479a3

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed
 

‎nbs/timegpt.ipynb

+2-1
Original file line numberDiff line numberDiff line change
@@ -2111,7 +2111,8 @@
21112111
"]\n",
21122112
"for hyp in hyps:\n",
21132113
" fcst_test = timegpt.forecast(df_train, h=12, **hyp)\n",
2114-
" fcst_test.insert(2, 'y', df_test['y'].values).sort_values(['unique_id', 'ds']).reset_index(drop=True)\n",
2114+
" fcst_test.insert(2, 'y', df_test['y'].values)\n",
2115+
" fcst_test = fcst_test.sort_values(['unique_id', 'ds']).reset_index(drop=True)\n",
21152116
" fcst_cv = timegpt.cross_validation(df_, h=12, **hyp)\n",
21162117
" fcst_cv = fcst_cv.sort_values(['unique_id', 'ds']).reset_index(drop=True)\n",
21172118
" pd.testing.assert_frame_equal(\n",

0 commit comments

Comments
 (0)