Forgot pdb statement

timkpaine · Oct 26, 2017 · e32aa9d · e32aa9d
1 parent dc1b330
commit e32aa9d
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 1 deletion.
diff --git a/lantern/extensions/hideinput/extension.py b/lantern/extensions/hideinput/extension.py
@@ -6,4 +6,3 @@ def load_jupyter_server_extension(nb_server_app):
     exporter_map['pdf_hidecode'] = PDFHideCodeExporter
     exporter_map['html_hidecode'] = HTMLHideCodeExporter
     print('Installing lantern nbconvert endpoints')
-    import pdb; pdb.set_trace()
diff --git a/lantern/plotting/plot_matplotlib.py b/lantern/plotting/plot_matplotlib.py
@@ -236,6 +236,9 @@ def plot(data, **kwargs):
             _MFA[0].set_ylabel(kwargs.get('ylabel'))
         if 'title' in kwargs:
             plt.title(kwargs.get('title'))
+
+        if kwargs.get('raw', False):
+            return plt
         return plt.show()
 
     @staticmethod