renormalize the plots value axis on every update

python-adaptive · Apr 23, 2019 · 5fa2dd4 · 5fa2dd4
1 parent 4889ab5
commit 5fa2dd4
Show file tree

Hide file tree

Showing 2 changed files with 15 additions and 6 deletions.
diff --git a/adaptive/learner/balancing_learner.py b/adaptive/learner/balancing_learner.py
@@ -279,7 +279,7 @@ def plot_function(*args):
         dm = dm.redim.values(**d)
 
         if dynamic:
-            return dm
+            return dm.map(lambda obj: obj.opts(framewise=True), hv.Element)
         else:
             # XXX: change when https://github.com/ioam/holoviews/issues/3085
             # is fixed.

diff --git a/adaptive/notebook_integration.py b/adaptive/notebook_integration.py
@@ -84,7 +84,8 @@ def in_ipynb():
 active_plotting_tasks = dict()
 
 
-def live_plot(runner, *, plotter=None, update_interval=2, name=None):
+def live_plot(runner, *, plotter=None, update_interval=2,
+              name=None, normalize=True):
     """Live plotting of the learner's data.
 
     Parameters
@@ -99,15 +100,19 @@ def live_plot(runner, *, plotter=None, update_interval=2, name=None):
         Name for the `live_plot` task in `adaptive.active_plotting_tasks`.
         By default the name is None and if another task with the same name
         already exists that other `live_plot` is canceled.
+    normalize : bool
+        Normalize (scale to fit) the frame upon each update.
 
     Returns
     -------
     dm : `holoviews.core.DynamicMap`
         The plot that automatically updates every `update_interval`.
     """
     if not _holoviews_enabled:
-        raise RuntimeError("Live plotting is not enabled; did you run "
-                           "'adaptive.notebook_extension()'?")
+        raise RuntimeError(
+            "Live plotting is not enabled; did you run "
+            "'adaptive.notebook_extension()'?"
+        )
 
     import holoviews as hv
     import ipywidgets
@@ -123,8 +128,12 @@ def plot_generator():
             else:
                 yield plotter(runner.learner)
 
-    dm = hv.DynamicMap(plot_generator(),
-                       streams=[hv.streams.Stream.define('Next')()])
+    steams = [hv.streams.Stream.define("Next")()]
+    dm = hv.DynamicMap(plot_generator(), streams=streams)
+
+    if normalize:
+        dm = dm.map(lambda obj: obj.opts(framewise=True), hv.Element)
+
     cancel_button = ipywidgets.Button(description='cancel live-plot',
                                       layout=ipywidgets.Layout(width='150px'))