BUG: Lazy load colormap through _manager.acquire() in merge

corteva · Feb 28, 2022 · 1ac7417 · 1ac7417
1 parent 033668f
commit 1ac7417
Show file tree

Hide file tree

Showing 2 changed files with 14 additions and 6 deletions.
diff --git a/docs/history.rst b/docs/history.rst
@@ -3,6 +3,7 @@ History
 
 Latest
 -------
+- BUG: Lazy load colormap through _manager.acquire() in merge (issue #479)
 
 0.10.1
 -------

diff --git a/rioxarray/merge.py b/rioxarray/merge.py
@@ -35,19 +35,26 @@ def __init__(self, xds: DataArray):
         try:
             rio_file = xds.rio._manager.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 colormap(self, *args, **kwargs):
+        """
+        Lazy load colormap through _manager.acquire()
+        for the scenario many file handles are opened
+
+        See: https://github.com/corteva/rioxarray/issues/479
+        """
+        try:
+            rio_file = self.xds.rio._manager.acquire()
+            return rio_file.colormap(*args, **kwargs)
+        except AttributeError:
+            return None
 
     def read(self, window, out_shape, *args, **kwargs) -> numpy.ma.array:
         # pylint: disable=unused-argument