Preserve encoding while making ROI (#111)

* Preserve encoding while making ROI * Fix for DataArrays
fmaussion · Jul 24, 2018 · 53b43ce · 53b43ce
1 parent d9de663
commit 53b43ce
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/salem/sio.py b/salem/sio.py
@@ -508,8 +508,12 @@ def roi(self, ds=None, **kwargs):
 
         out = self._obj.where(mask, other=other)
 
-        # keep attrs
+        # keep attrs and encoding
         out.attrs = self._obj.attrs
+        out.encoding = self._obj.encoding
+        if isinstance(out, xr.Dataset):
+            for v in self._obj.variables.keys():
+                out[v].encoding = self._obj[v].encoding
         if isinstance(out, xr.DataArray):
             out.name = self._obj.name