Plotting commands now return the plots they generate.

[calibcant.git] / calibcant / vibration_analyze.py

diff --git a/calibcant/vibration_analyze.py b/calibcant/vibration_analyze.py
index 8959d372bce29d93bf9aed4afe7ee7657122791e..0cfb27ca4ff8d4c72a4178750ecdba04e25d3ad6 100644 (file)
--- a/calibcant/vibration_analyze.py
+++ b/calibcant/vibration_analyze.py
@@ -559,4 +559,5 @@ def plot(deflection=None, freq_axis=None, power=None, A=None, B=None,
         freq_axes.set_ylabel('PSD')
     if hasattr(figure, 'show'):
         figure.show()
+    return figure
 _plot = plot  # alternative name for use inside analyze()