Fix notebook dependencies & enable videos in Colab

Author: nkarpachev

week1_09_approx_qlearning/week09_approximate_Q_learning.ipynb

@@ -24,7 +24,6 @@
     "# os.system('apt-get install -y xvfb')\n",
     "# os.system('wget https://raw.githubusercontent.com/yandexdataschool/Practical_DL/fall18/xvfb -O ../xvfb')\n",
     "# os.system('apt-get install -y python-opengl ffmpeg')\n",
-    "# os.system('pip install pyglet==1.2.4')\n",
     "\n",
     "# XVFB will be launched if you run on a server\n",
     "import os\n",
@@ -380,18 +379,30 @@
    "metadata": {},
    "outputs": [],
    "source": [
-    "# show video\n",
+    "# Show video. This may not work in some setups. If it doesn't\n",
+    "# work for you, you can download the videos and view them locally.\n",
+    "\n",
+    "import sys\n",
+    "from pathlib import Path\n",
+    "from base64 import b64encode\n",
     "from IPython.display import HTML\n",
-    "import os\n",
     "\n",
-    "video_names = list(\n",
-    "    filter(lambda s: s.endswith(\".mp4\"), os.listdir(\"./videos/\")))\n",
+    "video_paths = sorted([s for s in Path('videos').iterdir() if s.suffix == '.mp4'])\n",
+    "video_path = video_paths[-3]  # You can also try other indices\n",
+    "\n",
+    "if 'google.colab' in sys.modules:\n",
+    "    # https://stackoverflow.com/a/57378660/1214547\n",
+    "    with video_path.open('rb') as fp:\n",
+    "        mp4 = fp.read()\n",
+    "    data_url = 'data:video/mp4;base64,' + b64encode(mp4).decode()\n",
+    "else:\n",
+    "    data_url = str(video_path)\n",
     "\n",
     "HTML(\"\"\"\n",
     "<video width=\"640\" height=\"480\" controls>\n",
     "  <source src=\"{}\" type=\"video/mp4\">\n",
     "</video>\n",
-    "\"\"\".format(\"./videos/\"+video_names[-1]))  # this may or may not be _last_ video. Try other indices"
+    "\"\"\".format(data_url))"
    ]
   },
   {
@@ -405,9 +416,9 @@
  "metadata": {
   "anaconda-cloud": {},
   "kernelspec": {
-   "display_name": "Py3 research env",
+   "display_name": "Python 3",
    "language": "python",
-   "name": "py3_research"
+   "name": "python3"
   },
   "language_info": {
    "codemirror_mode": {
@@ -419,7 +430,7 @@
    "name": "python",
    "nbconvert_exporter": "python",
    "pygments_lexer": "ipython3",
-   "version": "3.6.7"
+   "version": "3.7.3"
   }
  },
  "nbformat": 4,