requested changes

microsoft · Feb 25, 2022 · 40b141f · 40b141f
1 parent b01bd80
commit 40b141f
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 5 deletions.
diff --git a/docs/api/datasets.rst b/docs/api/datasets.rst
@@ -47,8 +47,8 @@ Esri2020
 
 .. autoclass:: Esri2020
 
-EUDEM
-^^^^^
+EU-DEM
+^^^^^^
 
 .. autoclass:: EUDEM
 

diff --git a/torchgeo/datasets/eudem.py b/torchgeo/datasets/eudem.py
@@ -9,14 +9,13 @@
 
 import matplotlib.pyplot as plt
 from rasterio.crs import CRS
-from torch import Tensor
 
 from .geo import RasterDataset
 from .utils import check_integrity, extract_archive
 
 
 class EUDEM(RasterDataset):
-    """EUDEM Dataset.
+    """European Digital Elevation Model (EU-DEM) Dataset.
 
     The `EUDEM
     <https://land.copernicus.eu/imagery-in-situ/eu-dem/eu-dem-v1.1?tab=mapview>`_
@@ -142,7 +141,7 @@ def _verify(self) -> None:
 
     def plot(  # type: ignore[override]
         self,
-        sample: Dict[str, Tensor],
+        sample: Dict[str, Any],
         show_titles: bool = True,
         suptitle: Optional[str] = None,
     ) -> plt.Figure: