Serialize color table dict as yaml

rioxarray/_io.py

@@ -14,6 +14,7 @@
 
 import numpy as np
 import rasterio
+import yaml
 from packaging import version
 from rasterio.errors import NotGeoreferencedWarning
 from rasterio.vrt import WarpedVRT
@@ -535,7 +536,7 @@ def _get_rasterio_attrs(riods):
  if hasattr(riods, "colormap"):
  # A dict containing the colormap for a band
  try:
- attrs["colormap"] = riods.colormap(1)
+ attrs["colormap"] = yaml.safe_dump(riods.colormap(1))
  except ValueError: # NULL color table
  pass
  if hasattr(riods, "colorinterp") and any(riods.colorinterp):

rioxarray/raster_writer.py

@@ -12,6 +12,7 @@
 
 import numpy
 import rasterio
+import yaml
 from rasterio.windows import Window
 from xarray.conventions import encode_cf_variable
 
@@ -60,7 +61,7 @@ def _write_metatata_to_raster(raster_handle, xarray_dataset, tags):
  # see https://rasterio.readthedocs.io/en/latest/topics/color.html
  try:
  raster_handle.colorinterp = tags["colorinterp"]
- colormap = tags["colormap"]
+ colormap = yaml.safe_load(tags["colormap"])
  raster_handle.write_colormap(1, colormap)
  except KeyError:
  pass

test/integration/test_integration__io.py

@@ -1052,7 +1052,7 @@ def test_geotiff_attr_loading(open_rasterio):
  assert rds.shape == (1, 180, 360)
  # assert rds.attrs["scale_factor"] == 1.0
  # assert rds.attrs["add_offset"] == 0.0
- assert isinstance(rds.attrs["colormap"], dict)
+ assert isinstance(rds.attrs["colormap"], str)
  assert rds.attrs["colorinterp"] == (rasterio.enums.ColorInterp.palette,)