Merge branch 'master' into 'master'

crop_tgrid fixes

See merge request m300602/pyicon!23

pyicon/pyicon_calc_xr.py

@@ -232,6 +232,11 @@ def xr_crop_tgrid(ds_tg, ireg_c, verbose=1):
     ireg_e, inde = np.unique(edge_of_cell, return_index=True)
     ireg_v, indv = np.unique(vertex_of_cell, return_index=True)
 
+    ireg_c = ireg_c.astype("int32")
+    ireg_e = ireg_e.astype("int32")
+    ireg_v = ireg_v.astype("int32")
+    
+    
     # --- new dataset with cutted coordinates
     _print_verbose(verbose, "cut coordinates")
     ds_tg_cut = xr.Dataset(coords=dict(
@@ -257,37 +262,37 @@ def xr_crop_tgrid(ds_tg, ireg_c, verbose=1):
 
     var = 'vertex_of_cell'
     da = ds_tg[var].isel(cell=ireg_c) - offset
-    data = reindex_v[da.data.flatten()].reshape(da.shape)
+    data = reindex_v[da.data.flatten().compute().astype("int32")].reshape(da.shape)
     ds_tg_cut[var] = xr.DataArray(data + offset, dims=da.dims)
 
     var = "edge_vertices"
     da = ds_tg[var].isel(edge=ireg_e) - offset
-    data = reindex_v[da.data.flatten()].reshape(da.shape)
+    data = reindex_v[da.data.flatten().compute().astype("int32")].reshape(da.shape)
     ds_tg_cut[var] = xr.DataArray(data + offset, dims=da.dims)
 
     var = 'vertices_of_vertex'
     da = ds_tg[var].isel(vertex=ireg_v) - offset
-    data = reindex_v[da.data.flatten()].reshape(da.shape)
+    data = reindex_v[da.data.flatten().compute().astype("int32")].reshape(da.shape)
     ds_tg_cut[var] = xr.DataArray(data + offset, dims=da.dims)
 
     var = 'edge_of_cell'
     da = ds_tg[var].isel(cell=ireg_c) - offset
-    data = reindex_e[da.data.flatten()].reshape(da.shape)
+    data = reindex_e[da.data.flatten().compute().astype("int32")].reshape(da.shape)
     ds_tg_cut[var] = xr.DataArray(data + offset, dims=da.dims)
 
     var = 'edges_of_vertex'
     da = ds_tg[var].isel(vertex=ireg_v) - offset
-    data = reindex_e[da.data.flatten()].reshape(da.shape)
+    data = reindex_e[da.data.flatten().compute().astype("int32")].reshape(da.shape)
     ds_tg_cut[var] = xr.DataArray(data + offset, dims=da.dims)
 
     var = 'adjacent_cell_of_edge'
     da = ds_tg[var].isel(edge=ireg_e) - offset
-    data = reindex_c[da.data.flatten()].reshape(da.shape)
+    data = reindex_c[da.data.flatten().compute().astype("int32")].reshape(da.shape)
     ds_tg_cut[var] = xr.DataArray(data + offset, dims=da.dims)
 
     var = 'cells_of_vertex'
     da = ds_tg[var].isel(vertex=ireg_v) - offset
-    data = reindex_c[da.data.flatten()].reshape(da.shape)
+    data = reindex_c[da.data.flatten().compute().astype("int32")].reshape(da.shape)
     ds_tg_cut[var] = xr.DataArray(data + offset, dims=da.dims)
 
     reindex_vars = [