fix xarray get

ecmwf · Mar 6, 2024 · bce6fb4 · bce6fb4
1 parent ef80bb5
commit bce6fb4
Showing 1 changed file with 7 additions and 3 deletions.
diff --git a/polytope/datacube/backends/xarray.py b/polytope/datacube/backends/xarray.py
@@ -46,13 +46,17 @@ def get(self, requests: IndexTree):
                 path_copy = deepcopy(path)
                 for key in path_copy:
                     axis = self._axes[key]
-                    (path, unmapped_path) = axis.unmap_to_datacube(path, unmapped_path)
-                # TODO: here do nearest point search
-                path = self.fit_path(path)
+                    key_value_path = {key: path_copy[key]}
+                    (key_value_path, path, unmapped_path) = axis.unmap_path_key(key_value_path, path, unmapped_path)
+                path.update(key_value_path)
+
+                unmapped_path = {}
+                self.refit_path(path, unmapped_path, path)
                 for key in path:
                     path[key] = list(path[key])
                 for key in unmapped_path:
                     unmapped_path[key] = list(unmapped_path[key])
+
                 subxarray = self.dataarray.sel(path, method="nearest")
                 subxarray = subxarray.sel(unmapped_path)
                 value = subxarray.values