From 2268299987f0f6461a6a693bd0e78fe94a76ef4a Mon Sep 17 00:00:00 2001
From: isaaccorley <22203655+isaaccorley@users.noreply.github.com>
Date: Wed, 10 Nov 2021 17:58:52 -0600
Subject: [PATCH] style fixes

---
 torchgeo/datasets/utils.py | 2 +-
 torchgeo/datasets/xview.py | 2 +-
 2 files changed, 2 insertions(+), 2 deletions(-)

diff --git a/torchgeo/datasets/utils.py b/torchgeo/datasets/utils.py
index 099936ff3b7..0327378d16c 100644
--- 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
index fb405165beb..22a5fc85796 100644
--- 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(