From 62a91370f8f7be96ff2ddd6557c14ceac03fa734 Mon Sep 17 00:00:00 2001
From: Rishabh Raj <rishabhr123@gmail.com>
Date: Sun, 13 Apr 2014 15:08:28 +0530
Subject: [PATCH] Add __all__ to specify modules which should be imported

---
 doc/ext/notebook.py | 2 ++
 doc/ext/plot2rst.py | 2 +-
 2 files changed, 3 insertions(+), 1 deletion(-)

diff --git a/doc/ext/notebook.py b/doc/ext/notebook.py
index b62401fa..43ede80c 100644
--- a/doc/ext/notebook.py
+++ b/doc/ext/notebook.py
@@ -1,3 +1,5 @@
+__all__ = ['python_to_notebook']
+
 import json
 import copy
 
diff --git a/doc/ext/plot2rst.py b/doc/ext/plot2rst.py
index e951729f..07b54470 100644
--- a/doc/ext/plot2rst.py
+++ b/doc/ext/plot2rst.py
@@ -80,7 +80,7 @@ from skimage import io
 from skimage import transform
 from skimage.util.dtype import dtype_range
 
-from notebook import save_ipython_notebook
+from notebook import python_to_notebook
 
 
 LITERALINCLUDE = """