Merge 8a05ec9 into 60c4552

corteva · Jan 26, 2021 · e4fcdd8 · e4fcdd8
2 parents 60c4552 + 8a05ec9
commit e4fcdd8
Show file tree

Hide file tree

Showing 2 changed files with 17 additions and 0 deletions.
diff --git a/docs/history.rst b/docs/history.rst
@@ -7,6 +7,7 @@ Latest
 - ENH: Add support for merging datasets with different CRS (issue #173)
 - ENH: Add support for using dask in `rio.to_raster` (issue #9)
 - BUG: Allow `rio.write_crs` when spatial dimensions not found (pull #186)
+- BUG: Update to support rasterio 1.2+ merge (issue #180)
 
 0.1.1
 ------

diff --git a/rioxarray/merge.py b/rioxarray/merge.py
@@ -32,6 +32,22 @@ def __init__(self, xds: DataArray):
         res = xds.rio.resolution(recalc=True)
         self.res = (abs(res[0]), abs(res[1]))
         self.transform = xds.rio.transform(recalc=True)
+        try:
+            rio_file = xds._file_obj.acquire()
+            self.profile = rio_file.profile
+            self.colormap = rio_file.colormap
+        except AttributeError:
+            self.profile = {}
+
+            def colormap(_):
+                return None
+
+            self.colormap = colormap
+        self.profile.update(
+            dtype=xds.dtype,
+            crs=xds.rio.crs,
+            nodata=xds.rio.nodata,
+        )
 
     def read(self, window, out_shape, *args, **kwargs) -> numpy.ma.array:
         # pylint: disable=unused-argument