SustainBenchCropYield: fix download (#1755)

microsoft · Mar 2, 2024 · 348ae7c · 348ae7c
1 parent d147be1
commit 348ae7c
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/torchgeo/datasets/sustainbench_crop_yield.py b/torchgeo/datasets/sustainbench_crop_yield.py
@@ -223,7 +223,7 @@ def _download(self) -> None:
         download_url(
             self.url,
             self.root,
-            filename=self.dir,
+            filename=self.dir + ".zip",
             md5=self.md5 if self.checksum else None,
         )
         self._extract()