Ensure 'data_type' attribute is a string to make it valid for the dat…

…aset. Fixes #49. PiperOrigin-RevId: 571994074
google · Oct 9, 2023 · eb39a84 · eb39a84
1 parent 744491f
commit eb39a84
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/xee/ext.py b/xee/ext.py
@@ -331,6 +331,9 @@ def open_store_variable(self, name: str) -> xarray.Variable:
     x_dim_name, y_dim_name = self.dimension_names
     dimensions = [self.primary_dim_name, x_dim_name, y_dim_name]
     attrs = self._band_attrs(name)
+    # Convert 'data_type' attribute to string to make it valid for the dataset.
+    if attrs.get('data_type'):
+      attrs['data_type'] = str(attrs['data_type'])
     encoding = {
         'source': attrs['id'],
         'scale_factor': arr.scale,