From c05a84a9df46518860f783d3db7a8bc0143495ee Mon Sep 17 00:00:00 2001
From: Jake VanderPlas <jakevdp@gmail.com>
Date: Thu, 1 Dec 2016 10:00:10 -0800
Subject: [PATCH] Add kernelspec tool

---
 tools/fix_kernelspec.py | 19 +++++++++++++++++++
 1 file changed, 19 insertions(+)
 create mode 100644 tools/fix_kernelspec.py

diff --git a/tools/fix_kernelspec.py b/tools/fix_kernelspec.py
new file mode 100644
index 000000000..c98d5e927
--- /dev/null
+++ b/tools/fix_kernelspec.py
@@ -0,0 +1,19 @@
+import os
+
+import nbformat
+
+from generate_contents import iter_notebooks, NOTEBOOK_DIR
+
+def fix_kernelspec():
+    for nb_name in iter_notebooks():
+        nb_file = os.path.join(NOTEBOOK_DIR, nb_name)
+        nb = nbformat.read(nb_file, as_version=4)
+
+        print("- Updating kernelspec for {0}".format(nb_name))
+        nb['metadata']['kernelspec']['display_name'] = 'Python 3'
+
+        nbformat.write(nb, nb_file)
+
+
+if __name__ == '__main__':
+    fix_kernelspec()