code: def set_grid_attrs_as_coords(ds):
      
      grid_attrs_in_ds = set(GRID_ATTRS.keys()).intersection(
          set(ds.coords) | set(ds.data_vars))
      ds = ds.set_coords(grid_attrs_in_ds)
      return ds
docs: Set available grid attributes as coordinates in a given Dataset.
  
      Grid attributes are assumed to have their internal aospy names. Grid
      attributes are set as coordinates, such that they are carried by all
      selected DataArrays with overlapping index dimensions.
  
      Parameters
      ----------
      ds : Dataset
          Input data
  
      Returns
      -------
      Dataset
          Dataset with grid attributes set as coordinates