fix: replace deprecated plt.cm.get_cmap in colab notebook to fix CI

Author: elkins

examples/ml_integration/plm_embeddings.ipynb

@@ -479,7 +479,7 @@
     "\n",
     "    # Colour by amino acid identity\n",
     "    unique_aas = sorted(set(labels))\n",
-    "    cmap = plt.cm.get_cmap(\"tab20\", len(unique_aas))\n",
+    "    cmap = plt.get_cmap(\"tab20\", len(unique_aas))\n",
     "    color_map = {aa: cmap(i) for i, aa in enumerate(unique_aas)}\n",
     "    colors = [color_map[aa] for aa in labels]\n",
     "\n",
@@ -517,7 +517,7 @@
     "    ax = axes[1]\n",
     "    source_labels = np.array(all_source)\n",
     "    unique_sources = list(sequences.keys())\n",
-    "    scmap = plt.cm.get_cmap(\"Set2\", len(unique_sources))\n",
+    "    scmap = plt.get_cmap(\"Set2\", len(unique_sources))\n",
     "    for i, src in enumerate(unique_sources):\n",
     "        mask = source_labels == src\n",
     "        ax.scatter(\n",