File "/home/sxy26921/myruns/czi/01/../../../mygit/affinity-vae/run.py", line 693, in run
    run_pipeline(data)
  File "/home/sxy26921/myruns/czi/01/../../../mygit/affinity-vae/run.py", line 702, in run_pipeline
    train(
  File "/home/sxy26921/mygit/affinity-vae/avae/train.py", line 197, in train
    trains, vals, tests, affinity_matrix, data_dim = load_data(
  File "/home/sxy26921/mygit/affinity-vae/avae/data.py", line 169, in load_data
    affinity = get_affinity_matrix(affinity_path, classes_list)
  File "/home/sxy26921/mygit/affinity-vae/avae/data.py", line 282, in get_affinity_matrix
    plot_affinity_matrix(
  File "/home/sxy26921/mygit/affinity-vae/avae/vis.py", line 1890, in plot_affinity_matrix
    im = ax.imshow(lookup, vmin=-1, vmax=1, cmap=plt.cm.get_cmap("RdBu"))
AttributeError: module 'matplotlib.cm' has no attribute 'get_cmap'