From 5006164fb5d07a4496071a9c79c84274cf9ad33b Mon Sep 17 00:00:00 2001
From: jlstevens <jlstevens@ed.ac.uk>
Date: Fri, 14 Apr 2017 13:10:06 +0100
Subject: [PATCH] Removed max_branches argument from display hooks

---
 holoviews/ipython/display_hooks.py | 15 +++++++--------
 1 file changed, 7 insertions(+), 8 deletions(-)

diff --git a/holoviews/ipython/display_hooks.py b/holoviews/ipython/display_hooks.py
index 396db572e1..0fad09ef5f 100644
--- a/holoviews/ipython/display_hooks.py
+++ b/holoviews/ipython/display_hooks.py
@@ -129,8 +129,7 @@ def wrapped(element):
 
         try:
             html = fn(element,
-                      max_frames=OutputMagic.options['max_frames'],
-                      max_branches = None)
+                      max_frames=OutputMagic.options['max_frames'])
 
             # Only want to add to the archive for one display hook...
             disabled_suffixes = ['png_display', 'svg_display']
@@ -162,7 +161,7 @@ def wrapped(element):
 
 
 @display_hook
-def element_display(element, max_frames, max_branches):
+def element_display(element, max_frames):
     info = process_object(element)
     if info:
         IPython.display.display(IPython.display.HTML(info))
@@ -181,7 +180,7 @@ def element_display(element, max_frames, max_branches):
 
 
 @display_hook
-def map_display(vmap, max_frames, max_branches):
+def map_display(vmap, max_frames):
     if not isinstance(vmap, (HoloMap, DynamicMap)): return None
     if len(vmap) == 0 and (not isinstance(vmap, DynamicMap) or vmap.sampled):
         return sanitize_HTML(vmap)
@@ -193,7 +192,7 @@ def map_display(vmap, max_frames, max_branches):
 
 
 @display_hook
-def layout_display(layout, max_frames, max_branches):
+def layout_display(layout, max_frames):
     if isinstance(layout, AdjointLayout): layout = Layout.from_values(layout)
     if not isinstance(layout, (Layout, NdLayout)): return None
 
@@ -206,7 +205,7 @@ def layout_display(layout, max_frames, max_branches):
 
 
 @display_hook
-def grid_display(grid, max_frames, max_branches):
+def grid_display(grid, max_frames):
     if not isinstance(grid, GridSpace): return None
 
     nframes = len(unique_dimkeys(grid)[1])
@@ -252,7 +251,7 @@ def pprint_display(obj):
 
 
 @display_hook
-def element_png_display(element, max_frames, max_branches):
+def element_png_display(element, max_frames):
     """
     Used to render elements to PNG if requested in the display formats.
     """
@@ -277,7 +276,7 @@ def element_png_display(element, max_frames, max_branches):
 
 
 @display_hook
-def element_svg_display(element, max_frames, max_branches):
+def element_svg_display(element, max_frames):
     """
     Used to render elements to SVG if requested in the display formats.
     """