Questions about adding category annotation in pl.paga_path

[Yansr3]

Hello,

I'm trying to use plot heat map along trajectory with pl.paga_path function. Besides groups_key and pseudotime, I want to add another annotation "sample" in the heat map. However, with only four unique category value in the "sample" column, I got more than four colours in the heatmap annotation. I am wondering if there is any suggestions on adding category annotation in pl.paga_path heatmap.

sc.pl.paga_path(
        adata,
        path,
        ['GATA1','GATA2'],
        show_node_names=False,
        ytick_fontsize=12,
        left_margin=0.15,
        n_avg=50,
        annotations=["dpt_pseudotime","sample"],
        show_yticks=True,
        show_colorbar=False,
        color_map="inferno",
        groups_key="cell_type",
        color_maps_annotations={"dpt_pseudotime": "viridis", "sample": "tab20"},
        normalize_to_zero_one = True
    )