fix missing libraries and imports

Author: cbouy

docs/notebooks/callbacks.ipynb

@@ -16,7 +16,7 @@
    "outputs": [],
    "source": [
     "# uncomment and run if you're on Google Colab\n",
-    "# !pip install rdkit-pypi mols2grid\n",
+    "# !pip install rdkit-pypi mols2grid py3Dmol\n",
     "# !wget https://raw.githubusercontent.com/rdkit/rdkit/master/Docs/Book/data/solubility.test.sdf"
    ]
   },
@@ -30,7 +30,8 @@
     "from pathlib import Path\n",
     "from rdkit import RDConfig\n",
     "from ipywidgets import widgets\n",
-    "import urllib\n",
+    "import urllib.request, urllib.parse\n",
+    "from urllib.error import HTTPError\n",
     "from IPython.display import display\n",
     "import py3Dmol\n",
     "SDF_FILE = (f\"{RDConfig.RDDocsDir}/Book/data/solubility.test.sdf\"\n",
@@ -71,6 +72,7 @@
     "# its presence is mandatory if you want to print/display some info with your callback\n",
     "@output.capture(clear_output=True, wait=True)\n",
     "def show_data(data):\n",
+    "    data.pop(\"img\")\n",
     "    print(data)\n",
     "\n",
     "view = mols2grid.display(\n",
@@ -109,7 +111,7 @@
     "    smi = urllib.parse.quote(data[\"SMILES\"])\n",
     "    try:\n",
     "        response = urllib.request.urlopen(url.format(smi))\n",
-    "    except urllib.error.HTTPError:\n",
+    "    except HTTPError:\n",
     "        print(f\"Could not find corresponding match on PubChem\")\n",
     "        print(data[\"SMILES\"])\n",
     "    else:\n",