Oops, add output argument to pypiezo.wiggle._setup_plot().

diff --git a/pypiezo/wiggle.py b/pypiezo/wiggle.py
index 87658e8afb70ecc659304d073c08956fd21f6974..06b47f476bec7bc0e4e7f16e5f4848be0d43478b 100644 (file)
--- a/pypiezo/wiggle.py
+++ b/pypiezo/wiggle.py
@@ -127,7 +127,7 @@ def _update_datafile(filename, group, config, cycle, data):
         cwg['time'] = timestamp
         cwg['raw/{}'.format(config['input'])] = data
 
-def _setup_plot(piezo, config):
+def _setup_plot(piezo, config, output):
     if not _matplotlib:
         raise _matplotlib_import_error
     _matplotlib.interactive(True)
@@ -136,7 +136,7 @@ def _setup_plot(piezo, config):
     axes.hold(False)
     timestamp = _time.strftime('%H%M%S')
     axes.set_title('wiggle for interference %s' % timestamp)
-    plot = axes.plot(out, out, 'b.-')
+    plot = axes.plot(output, output, 'b.-')
     figure.show()
     _matplotlib_pyplot.draw()
     _matplotlib_pyplot.show()