base recurrent import split

Nixtla · Nov 7, 2023 · d512fa7 · d512fa7
1 parent d5f5b92
commit d512fa7
Showing 1 changed file with 11 additions and 10 deletions.
diff --git a/nbs/common.base_recurrent.ipynb b/nbs/common.base_recurrent.ipynb
@@ -691,11 +691,19 @@
    "outputs": [],
    "source": [
     "#| hide\n",
-    "# add h=0,1 unit test for _parse_windows \n",
     "from neuralforecast.losses.pytorch import MAE\n",
     "from neuralforecast.utils import AirPassengersDF\n",
-    "from neuralforecast.tsdataset import TimeSeriesDataset, TimeSeriesDataModule\n",
-    "\n",
+    "from neuralforecast.tsdataset import TimeSeriesDataset, TimeSeriesDataModule"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "#| hide\n",
+    "# add h=0,1 unit test for _parse_windows \n",
     "# Declare batch\n",
     "AirPassengersDF['x'] = np.array(len(AirPassengersDF))\n",
     "AirPassengersDF['x2'] = np.array(len(AirPassengersDF)) * 2\n",
@@ -729,13 +737,6 @@
     "test_eq(set(temporal_data_cols), set(['y', 'x', 'x2']))\n",
     "test_eq(windows['temporal'].shape, torch.Size([1,len(['y', 'x', 'x2', 'available_mask']),117,12+1]))"
    ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": null,
-   "metadata": {},
-   "outputs": [],
-   "source": []
   }
  ],
  "metadata": {