ruff

microsoft · Dec 8, 2024 · d845385 · d845385
1 parent c686170
commit d845385
Showing 1 changed file with 13 additions and 8 deletions.
diff --git 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",
+    ")"
    ]
   },
   {