style fixes

microsoft · Nov 10, 2021 · 2268299 · 2268299
1 parent 7829917
commit 2268299
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/torchgeo/datasets/utils.py b/torchgeo/datasets/utils.py
@@ -33,7 +33,7 @@
     "collate_dict",
     "rasterio_loader",
     "dataset_split",
-    "draw_semantic_segmentation_masks"
+    "draw_semantic_segmentation_masks",
 )
 
 

diff --git a/torchgeo/datasets/xview.py b/torchgeo/datasets/xview.py
@@ -178,7 +178,7 @@ def plot(self, index: int, alpha: float = 0.5) -> plt.Figure:
             index: the index of the sample to plot
 
         Returns:
-
+            the matplotlib.pyplot figure
         """
         sample = self[index]
         image1 = draw_semantic_segmentation_masks(