From 1753b740eac7ae77db4985117acd6e6d0fef9cdc Mon Sep 17 00:00:00 2001 From: Jim Crist Date: Fri, 14 Aug 2015 18:52:10 -0500 Subject: [PATCH] Changes to profiler plotting - If `bokeh.plotting.output_notebook()` is called, don't write to file --- dask/diagnostics/profile_visualize.py | 7 +++++-- 1 file changed, 5 insertions(+), 2 deletions(-) diff --git a/dask/diagnostics/profile_visualize.py b/dask/diagnostics/profile_visualize.py index f5b6f39449d..82848088254 100644 --- a/dask/diagnostics/profile_visualize.py +++ b/dask/diagnostics/profile_visualize.py @@ -5,6 +5,7 @@ from toolz import unique, groupby import bokeh.plotting as bp +from bokeh.io import _state from bokeh.palettes import brewer from bokeh.models import HoverTool @@ -78,7 +79,7 @@ def get_colors(palette, funcs): return [color_lookup[n] for n in funcs] -def visualize(results, dsk, palette='GnBu', file_path="profile.html", +def visualize(results, dsk, palette='GnBu', file_path=None, show=True, **kwargs): """Visualize the results of profiling in a bokeh plot. @@ -103,7 +104,9 @@ def visualize(results, dsk, palette='GnBu', file_path="profile.html", The completed bokeh plot object. """ - bp.output_file(file_path) + if not _state._notebook: + file_path = file_path or "profile.html" + bp.output_file(file_path) keys, tasks, starts, ends, ids = zip(*results) id_group = groupby(itemgetter(4), results)