Sentinel 2 tiler mosaic grid coverage

[dvd3v]

I am running a TiTiler application on AWS Lambda with theSentinel scenesand Sentinel mosaic endpoints, based on the exact example from the docs: https://devseed.com/titiler/examples/code/tiler_for_sentinel2/

When running step 3 from the example, i'm not getting a fully covered mosaic. Here is an example with return_mask turned of:

When looking at the tile features I believe the problem is that not all tiles are shown when the grids overlap:

I've tried running it with and without the optimized_filter when creating the Mosaic file. Also the minimum_tile_cover and tile_cover_sort do not seem to have effect.

# Optimized Mosaic
def optimized_filter(
    tile: mercantile.Tile,  # noqa
    dataset: Sequence[Dict],
    geoms: Sequence[polygons],
    minimum_tile_cover=None,  # noqa
    tile_cover_sort=False,  # noqa
    maximum_items_per_tile: Optional[int] = None,
) -> List:
    """Optimized filter that keeps only one item per grid ID."""
    gridid: List[str] = []
    selected_dataset: List[Dict] = []

    for item in dataset:
        grid = item["id"].split("_")[1]
        if grid not in gridid:
            gridid.append(grid)
            selected_dataset.append(item)

    dataset = selected_dataset

    indices = list(range(len(dataset)))
    if maximum_items_per_tile:
        indices = indices[:maximum_items_per_tile]

    return [dataset[ind] for ind in indices]


with MosaicBackend(
    "stac+https://earth-search.aws.element84.com/v0/search",
    query,
    minzoom=5,
    maxzoom=14,
    mosaic_options={"accessor": custom_accessor, "asset_filter": optimized_filter},
) as mosaic:
    mosaic_doc = mosaic.mosaic_def

Is it possible to get a fully covered mosaic within the AOI, using the pixel_selection method to create the composite?