[pre-commit.ci] auto fixes from pre-commit.com hooks

for more information, see https://pre-commit.ci

Author: pre-commit-ci[bot]

docs/partial-inputs-flood-tutorial.ipynb

@@ -410,7 +410,7 @@
     "multi_model = CLAYModule.load_from_checkpoint(\n",
     "    CKPT_PATH,\n",
     "    mask_ratio=0.0,\n",
-    "    band_groups={\"rgb\": (2, 1, 0), \"nir\": (3,), \"swir\": (4,5)},\n",
+    "    band_groups={\"rgb\": (2, 1, 0), \"nir\": (3,), \"swir\": (4, 5)},\n",
     "    bands=6,\n",
     "    strict=False,  # ignore the extra parameters in the checkpoint\n",
     ")\n",
@@ -427,7 +427,6 @@
     "        2893.86,  # nir\n",
     "        2303.00,  # swir16\n",
     "        1807.79,  # swir22\n",
-    "        \n",
     "    ]\n",
     "    STD = [\n",
     "        2026.96,  # red\n",
@@ -472,7 +471,7 @@
     "        batch[\"pixels\"] = batch[\"pixels\"].to(multi_model.device)\n",
     "        # Pass just the specific band through the model\n",
     "        batch[\"timestep\"] = batch[\"timestep\"].to(multi_model.device)\n",
-    "        batch[\"date\"] = batch[\"date\"] #.to(multi_model.device)\n",
+    "        batch[\"date\"] = batch[\"date\"]  # .to(multi_model.device)\n",
     "        batch[\"latlon\"] = batch[\"latlon\"].to(multi_model.device)\n",
     "\n",
     "        # Pass pixels, latlon, timestep through the encoder to create encoded patches\n",
@@ -651,17 +650,17 @@
     "plt.xticks(rotation=-30)\n",
     "# All points\n",
     "plt.scatter(tss, pca_result, color=\"blue\")\n",
-    "#plt.scatter(stack.time, pca_result, color=\"blue\")\n",
+    "# plt.scatter(stack.time, pca_result, color=\"blue\")\n",
     "\n",
     "# Cloudy images\n",
     "plt.scatter(tss[7], pca_result[7], color=\"green\")\n",
     "plt.scatter(tss[8], pca_result[8], color=\"green\")\n",
-    "#plt.scatter(stack.time[7], pca_result[7], color=\"green\")\n",
-    "#plt.scatter(stack.time[8], pca_result[8], color=\"green\")\n",
+    "# plt.scatter(stack.time[7], pca_result[7], color=\"green\")\n",
+    "# plt.scatter(stack.time[8], pca_result[8], color=\"green\")\n",
     "\n",
     "# After flood\n",
     "plt.scatter(tss[-7:], pca_result[-7:], color=\"red\")\n",
-    "#plt.scatter(stack.time[-7:], pca_result[-7:], color=\"red\")"
+    "# plt.scatter(stack.time[-7:], pca_result[-7:], color=\"red\")"
    ]
   },
   {
@@ -691,7 +690,6 @@
    "outputs": [],
    "source": [
     "from sklearn.manifold import TSNE\n",
-    "from sklearn.ensemble import IsolationForest\n",
     "\n",
     "# Perform t-SNE on the embeddings\n",
     "tsne = TSNE(n_components=2, perplexity=5)\n",
@@ -722,11 +720,11 @@
     "\n",
     "# Annotate each point with the corresponding date\n",
     "for i, (x, y) in enumerate(zip(X_tsne[:, 0], X_tsne[:, 1])):\n",
-    "    plt.annotate(f'{tss[i]}', (x, y))\n",
-    "    \n",
-    "plt.title('t-SNE Visualization')\n",
-    "plt.xlabel('t-SNE Component 1')\n",
-    "plt.ylabel('t-SNE Component 2')\n",
+    "    plt.annotate(f\"{tss[i]}\", (x, y))\n",
+    "\n",
+    "plt.title(\"t-SNE Visualization\")\n",
+    "plt.xlabel(\"t-SNE Component 1\")\n",
+    "plt.ylabel(\"t-SNE Component 2\")\n",
     "plt.show()"
    ]
   },

src/datamodule.py

@@ -19,6 +19,7 @@
 os.environ["GDAL_DISABLE_READDIR_ON_OPEN"] = "EMPTY_DIR"
 os.environ["GDAL_HTTP_MERGE_CONSECUTIVE_RANGES"] = "YES"
 
+
 # %%
 # Regular torch Dataset
 class ClayDataset(Dataset):
@@ -60,7 +61,14 @@ def read_chip(self, chip_path):
 
         # read timestep & normalize
         date = chip.tags()["date"]  # YYYY-MM-DD
-        year, month, day, year_non_norm, month_non_norm, day_non_norm = self.normalize_timestamp(date)
+        (
+            year,
+            month,
+            day,
+            year_non_norm,
+            month_non_norm,
+            day_non_norm,
+        ) = self.normalize_timestamp(date)
 
         # read lat,lon from UTM to WGS84 & normalize
         bounds = chip.bounds  # xmin, ymin, xmax, ymax