nilearn plotting не отображает метки

Question

Я использую следующий код для построения корреляционной матрицы.

from nilearn import plotting
import matplotlib.pyplot as plt

def plotMatrix(matrix, plot_path, labels, title, ticks, vmin, vmax):
    """Plot matrix. 
    param matrix: two dimensional array. The matrix to plot
    param plot_path : string. Full path and name of the plotting picture
    param labels: list. The labels 
    param title: string. The title of the plot
    param ticks: list. The ticks of the plot
    vmin: float. Minimum value
    vmax: float. Maximum value
    return: None
    """
    ticks = list(map(lambda x: x - 0.5, ticks))
    ticks_middle = [(((ticks[i+1]-ticks[i]) / 2 ) + ticks[i]) for i in range(0,len(ticks)-1)]
    fig, ax = plt.subplots()
    fig.set_size_inches(16.5, 9.5)
    plt.yticks(ticks_middle,list(labels))
    plt.xticks(ticks_middle,list(labels), rotation = 55, horizontalalignment='right')
    ax.xaxis.set_minor_locator(plt.FixedLocator(ticks[1:]))
    ax.yaxis.set_minor_locator(plt.FixedLocator(ticks[1:]))
    ax.grid(color='black', linestyle='-', linewidth=1.2, which='minor')
    plt.title(label = title, fontsize = 20)
    for item in (ax.get_xticklabels() + ax.get_yticklabels()):
        item.set_color(labelToColorDic[item.get_text()])
        item.set_fontsize(14)
    #plt.show()
    plotting.plot_matrix(matrix, colorbar=True, axes = ax, vmin=vmin, vmax=vmax)
    fig.savefig(plot_path)
    plt.close()

Предполагается, что результат будет таким, как на следующем рисунке:Матрица корреляции с метками, но я получаю матрицу без меток. Если добавить plt.show() перед тем plotting.plot_matrix, я получаю пустую матрицу, но с ярлыками на, как на рисунке ниже:Пустые матрицы корреляции с этикетками. Этот код работал год назад, вероятно, это вызвано новыми версиями, но я понятия не имею, что вызвало разницу. Есть идеи? Я также открыт для предложений по построению корреляционной матрицы с нужными мне метками.