Oops, bump_analyze.plot should return `figure`, not `plot`.

diff --git a/calibcant/bump_analyze.py b/calibcant/bump_analyze.py
index d855909042294fed6593fee1460087e055a433bb..140add37fc05485bf6670751998e8265f23b00bd 100644 (file)
--- a/calibcant/bump_analyze.py
+++ b/calibcant/bump_analyze.py
@@ -402,5 +402,5 @@ def plot(data, yguess=None, yfit=None):
         residual_axes.set_ylabel('Photodiode (Volts)')
     if hasattr(figure, 'show'):
         figure.show()
-    return plot
+    return figure
 _plot = plot  # alternative name for use inside fit()