We will need to remove the GeoTransform attribute in `assign_index` and add a helper function to write it back (similar to `.rio.write_transform`)