From 468860acdd23471226d37f79f9fb23b8fcd7d26b Mon Sep 17 00:00:00 2001
From: Rishabh Raj <rishabhr123@gmail.com>
Date: Mon, 14 Apr 2014 04:49:19 +0530
Subject: [PATCH] Add class to __all__, remove duplicates now private

---
 doc/ext/notebook.py | 6 +++---
 1 file changed, 3 insertions(+), 3 deletions(-)

diff --git a/doc/ext/notebook.py b/doc/ext/notebook.py
index b3123a51..5c57fa16 100644
--- a/doc/ext/notebook.py
+++ b/doc/ext/notebook.py
@@ -1,4 +1,4 @@
-__all__ = ['python_to_notebook']
+__all__ = ['python_to_notebook', 'Notebook']
 
 import json
 import copy
@@ -29,7 +29,7 @@ sample = """{
 }"""
 
 
-def remove_consecutive_duplicates(inp):
+def _remove_consecutive_duplicates(inp):
     """Remove duplicates of elements appearing consecutively.
 
     Parameters
@@ -152,7 +152,7 @@ def python_to_notebook(example_file, notebook_dir, notebook_path):
     docstring = False
     source = []
 
-    modified_code = remove_consecutive_duplicates(nb.code)
+    modified_code = _remove_consecutive_duplicates(nb.code)
 
     for line in modified_code:
         # A linebreak indicates a segment has ended.