From d84538553c5d9c9a62ddd6c4a70d317582b53709 Mon Sep 17 00:00:00 2001
From: "Adam J. Stewart" <ajstewart426@gmail.com>
Date: Sun, 8 Dec 2024 09:02:34 -0500
Subject: [PATCH] ruff

---
 docs/tutorials/torchgeo.ipynb | 21 +++++++++++++--------
 1 file changed, 13 insertions(+), 8 deletions(-)

diff --git a/docs/tutorials/torchgeo.ipynb b/docs/tutorials/torchgeo.ipynb
index 4310e4d4f4e..8b2dd28a0b8 100644
--- a/docs/tutorials/torchgeo.ipynb
+++ b/docs/tutorials/torchgeo.ipynb
@@ -71,12 +71,9 @@
     "import os\n",
     "import tempfile\n",
     "\n",
-    "import kornia.augmentation as K\n",
-    "import torch\n",
-    "from torch import nn, optim\n",
     "from torch.utils.data import DataLoader\n",
     "\n",
-    "from torchgeo.datasets import BoundingBox, CDL, Landsat7, Landsat8, stack_samples\n",
+    "from torchgeo.datasets import CDL, BoundingBox, Landsat7, Landsat8, stack_samples\n",
     "from torchgeo.datasets.utils import download_url\n",
     "from torchgeo.samplers import GridGeoSampler, RandomGeoSampler"
    ]
@@ -147,8 +144,12 @@
     "download_url()\n",
     "download_url()\n",
     "\n",
-    "landsat7 = Landsat7(paths=landsat_root, bands=['B1', 'B2', 'B3', 'B4', 'B5', 'B6', 'B7'])\n",
-    "landsat8 = Landsat8(paths=landsat_root, bands=['B2', 'B3', 'B4', 'B5', 'B6', 'B7', 'B8'])\n",
+    "landsat7 = Landsat7(\n",
+    "    paths=landsat_root, bands=['B1', 'B2', 'B3', 'B4', 'B5', 'B6', 'B7']\n",
+    ")\n",
+    "landsat8 = Landsat8(\n",
+    "    paths=landsat_root, bands=['B2', 'B3', 'B4', 'B5', 'B6', 'B7', 'B8']\n",
+    ")\n",
     "\n",
     "print(landsat7)\n",
     "print(landsat8)"
@@ -368,8 +369,12 @@
    "metadata": {},
    "outputs": [],
    "source": [
-    "train_dataloader = DataLoader(dataset, batch_size=128, sampler=train_sampler, collate_fn=stack_samples)\n",
-    "test_dataloader = DataLoader(dataset, batch_size=128, sampler=test_sampler, collate_fn=stack_samples)"
+    "train_dataloader = DataLoader(\n",
+    "    dataset, batch_size=128, sampler=train_sampler, collate_fn=stack_samples\n",
+    ")\n",
+    "test_dataloader = DataLoader(\n",
+    "    dataset, batch_size=128, sampler=test_sampler, collate_fn=stack_samples\n",
+    ")"
    ]
   },
   {