diff --git a/nbs/022_tslearner.ipynb b/nbs/022_tslearner.ipynb
index f0520bd1..2061edb9 100644
--- a/nbs/022_tslearner.ipynb
+++ b/nbs/022_tslearner.ipynb
@@ -265,10 +265,10 @@
        "  <tbody>\n",
        "    <tr>\n",
        "      <td>0</td>\n",
-       "      <td>1.464314</td>\n",
-       "      <td>0.233333</td>\n",
-       "      <td>1.400173</td>\n",
-       "      <td>0.166667</td>\n",
+       "      <td>1.446255</td>\n",
+       "      <td>0.266667</td>\n",
+       "      <td>1.403359</td>\n",
+       "      <td>0.300000</td>\n",
        "      <td>00:00</td>\n",
        "    </tr>\n",
        "  </tbody>\n",
@@ -339,8 +339,8 @@
        "  <tbody>\n",
        "    <tr>\n",
        "      <td>0</td>\n",
-       "      <td>1.438072</td>\n",
-       "      <td>0.200000</td>\n",
+       "      <td>1.286023</td>\n",
+       "      <td>0.400000</td>\n",
        "      <td>00:00</td>\n",
        "    </tr>\n",
        "  </tbody>\n",
@@ -600,11 +600,11 @@
        "  <tbody>\n",
        "    <tr>\n",
        "      <td>0</td>\n",
-       "      <td>221.817291</td>\n",
-       "      <td>14.270400</td>\n",
-       "      <td>209.151230</td>\n",
-       "      <td>14.046944</td>\n",
-       "      <td>00:01</td>\n",
+       "      <td>221.239578</td>\n",
+       "      <td>14.241582</td>\n",
+       "      <td>208.787231</td>\n",
+       "      <td>14.034328</td>\n",
+       "      <td>00:00</td>\n",
        "    </tr>\n",
        "  </tbody>\n",
        "</table>"
@@ -619,9 +619,9 @@
    ],
    "source": [
     "X, y, splits = get_regression_data('AppliancesEnergy', split_data=False)\n",
-    "X = X.astype('float32')\n",
-    "y = y.astype('float32')\n",
     "if X is not None: # This is to prevent a test fail when the data server is not available\n",
+    "    X = X.astype('float32')\n",
+    "    y = y.astype('float32')\n",
     "    batch_tfms = [TSStandardize()]\n",
     "    learn = TSRegressor(X, y, splits=splits, batch_tfms=batch_tfms, arch=None, metrics=mae, bs=512, train_metrics=True, device=default_device())\n",
     "    learn.fit_one_cycle(1, 1e-4)"
@@ -860,10 +860,10 @@
        "  <tbody>\n",
        "    <tr>\n",
        "      <td>0</td>\n",
-       "      <td>4114.624023</td>\n",
-       "      <td>48.891418</td>\n",
-       "      <td>7991.095703</td>\n",
-       "      <td>74.791130</td>\n",
+       "      <td>4616.225098</td>\n",
+       "      <td>53.340523</td>\n",
+       "      <td>7969.317871</td>\n",
+       "      <td>74.670258</td>\n",
        "      <td>00:00</td>\n",
        "    </tr>\n",
        "  </tbody>\n",
@@ -932,9 +932,9 @@
      "name": "stdout",
      "output_type": "stream",
      "text": [
-      "/Users/nacho/notebooks/tsai/nbs/022_tslearner.ipynb saved at 2024-02-11 10:55:07\n",
+      "/Users/nacho/notebooks/tsai/nbs/022_tslearner.ipynb saved at 2024-02-11 13:56:13\n",
       "Correct notebook to script conversion! 😃\n",
-      "Sunday 11/02/24 10:55:10 CET\n"
+      "Sunday 11/02/24 13:56:16 CET\n"
      ]
     },
     {
diff --git a/nbs/models/test.pth b/nbs/models/test.pth
index fcf8ad2e..7cd3f29a 100644
Binary files a/nbs/models/test.pth and b/nbs/models/test.pth differ