update get_cmap to colormaps

seanrjohnson · Jun 2, 2023 · afad513 · afad513
1 parent f175994
commit afad513
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/colab_notebooks/PAEView.ipynb b/colab_notebooks/PAEView.ipynb
@@ -5,7 +5,7 @@
     "colab": {
       "name": "PAEView.ipynb",
       "provenance": [],
-      "authorship_tag": "ABX9TyMDHO5lixoYDgvQ/Ok2/IYr",
+      "authorship_tag": "ABX9TyO0VhvlnTOTRk4GJQ/m6kGR",
       "include_colab_link": true
     },
     "kernelspec": {
@@ -129,7 +129,7 @@
         "from IPython.display import display\n",
         "from ipywidgets import interact, interactive, fixed, interact_manual\n",
         "import py3Dmol\n",
-        "from matplotlib.cm import get_cmap\n",
+        "from matplotlib import colormaps\n",
         "from matplotlib.colors import to_hex\n",
         "import matplotlib as mpl\n",
         "import matplotlib.pyplot as plt\n",
@@ -181,7 +181,7 @@
         "  global pdb_to_json\n",
         "  global uploaded_files\n",
         "  global view\n",
-        "  pae_cmap = get_cmap(args[\"color_scheme\"])\n",
+        "  pae_cmap = colormaps[args[\"color_scheme\"]]\n",
         "  \n",
         "  view.removeAllModels()\n",
         "  view.addModel(uploaded_files[args[\"pdb_file\"]].decode(\"utf-8\"), 'pdb')\n",