From 348ae7ccf27c6c9ef8ce8bbe5369c128bf658e50 Mon Sep 17 00:00:00 2001
From: "Adam J. Stewart" <ajstewart426@gmail.com>
Date: Mon, 4 Dec 2023 10:30:51 +0100
Subject: [PATCH] SustainBenchCropYield: fix download (#1755)

---
 torchgeo/datasets/sustainbench_crop_yield.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/torchgeo/datasets/sustainbench_crop_yield.py b/torchgeo/datasets/sustainbench_crop_yield.py
index 5619697b932..07169fd98b6 100644
--- 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()