BUG: ensure band_key is list when iterating over bands for mask and s…

…cale (#87)
corteva · Feb 3, 2020 · 94ede70 · 94ede70
1 parent 3c361b8
commit 94ede70
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 1 deletion.
diff --git a/docs/history.rst b/docs/history.rst
@@ -1,6 +1,10 @@
 History
 =======
 
+0.0.20
+------
+- BUG: ensure band_key is list when iterating over bands for mask and scale (pull #87)
+
 0.0.19
 -------
 - Add support for writing scales & offsets to raster (pull #79)

diff --git a/rioxarray/_io.py b/rioxarray/_io.py
@@ -160,7 +160,7 @@ def _getitem(self, key):
                 if self.masked:
                     out = np.ma.filled(out.astype(self.dtype), np.nan)
                 if self.mask_and_scale:
-                    for band in band_key:
+                    for band in np.atleast_1d(band_key):
                         band_iii = band - 1
                         out[band_iii] = (
                             out[band_iii] * riods.scales[band_iii]