Merge branch 'plotCDL' of github.com:nilsleh/torchgeo into plotCDL

microsoft · Feb 28, 2022 · 05273e4 · 05273e4
2 parents cd9c87b + aa4a42b
commit 05273e4
Showing 1 changed file with 2 additions and 3 deletions.
diff --git a/torchgeo/datasets/cdl.py b/torchgeo/datasets/cdl.py
@@ -5,12 +5,11 @@
 
 import glob
 import os
-from typing import Any, Callable, Dict, Optional, Tuple
+from typing import Any, Callable, Dict, Optional
 
 import matplotlib.pyplot as plt
 import numpy as np
 from rasterio.crs import CRS
-from torch import Tensor
 
 from .geo import RasterDataset
 from .utils import download_url, extract_archive
@@ -401,7 +400,7 @@ def _extract(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: