Matplotlib colorbar update (#573)

encoder/inference.py

@@ -171,7 +171,8 @@ def plot_embedding_as_heatmap(embed, ax=None, title="", shape=None, color_range=
     cmap = cm.get_cmap()
     mappable = ax.imshow(embed, cmap=cmap)
     cbar = plt.colorbar(mappable, ax=ax, fraction=0.046, pad=0.04)
-    cbar.set_clim(*color_range)
+    sm = cm.ScalarMappable(cmap=cmap)
+    sm.set_clim(*color_range)
     
     ax.set_xticks([]), ax.set_yticks([])
     ax.set_title(title)

requirements.txt

@@ -2,7 +2,7 @@ tensorflow==1.15
 umap-learn
 visdom
 librosa>=0.8.0
-matplotlib>=2.0.2
+matplotlib>=3.3.0
 numpy>=1.14.0
 scipy>=1.0.0
 tqdm