Merge pull request #808 from Kaggle/papermill

pip install papermill

Author: djherbis

Dockerfile

@@ -422,6 +422,8 @@ RUN pip install flashtext && \
     pip install geopandas==0.6.3 && \
     pip install nnabla && \
     pip install vowpalwabbit && \
+    # papermill can replace nbconvert for executing notebooks
+    pip install papermill && \
     /tmp/clean-layer.sh
 
 # Tesseract and some associated utility packages

tests/data/notebook.ipynb

@@ -11,6 +11,7 @@
     }
   ],
   "metadata":{
+    "kernelspec":{"language":"python","display_name":"Python 3","name":"python3"},
     "language_info":{
        "codemirror_mode":{
           "name":"ipython",
@@ -26,4 +27,4 @@
   },
   "nbformat":4,
   "nbformat_minor":4
-}
+}

tests/test_papermill.py

@@ -0,0 +1,14 @@
+import unittest
+
+import subprocess
+
+class TestPapermill(unittest.TestCase):
+    def test_papermill(self):
+        result = subprocess.run([
+            'papermill',
+            '/input/tests/data/notebook.ipynb',
+            '-',
+        ], stdout=subprocess.PIPE)
+
+        self.assertEqual(0, result.returncode)
+        self.assertTrue(b'999' in result.stdout)