VP transform: every band needs the CRS attribute

opendatacube · Oct 8, 2019 · 9460653 · 9460653
1 parent 94dec9a
commit 9460653
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/datacube/virtual/impl.py b/datacube/virtual/impl.py
@@ -485,7 +485,12 @@ def group(self, datasets: VirtualDatasetBag, **group_settings: Dict[str, Any]) -
         return self._input.group(datasets, **group_settings)
 
     def fetch(self, grouped: VirtualDatasetBox, **load_settings: Dict[str, Any]) -> xarray.Dataset:
-        return self._transformation.compute(self._input.fetch(grouped, **load_settings))
+        input_data = self._input.fetch(grouped, **load_settings)
+        output_data = self._transformation.compute(input_data)
+        output_data.attrs['crs'] = input_data.attrs['crs']
+        for data_var in output_data.data_vars:
+            output_data[data_var].attrs['crs'] = input_data.attrs['crs']
+        return output_data
 
 
 class Aggregate(VirtualProduct):