Small fix for dynamic collation closure

holoviz · Apr 4, 2017 · 2c5372c · 2c5372c
1 parent d8b402e
commit 2c5372c
Showing 1 changed file with 4 additions and 3 deletions.
diff --git a/holoviews/core/spaces.py b/holoviews/core/spaces.py
@@ -847,9 +847,10 @@ def collation_cb(*args, **kwargs):
                     vstreams = streams.get(i, [])
                 elif isinstance(streams, list):
                     vstreams = streams[i] if i < len(streams) else []
-                def collation_cb(collation_key=k, *args, **kwargs):
-                    return self[args][collation_key]
-                callback = Callable(collation_cb, inputs=[self])
+                def collation_cb(*args, **kwargs):
+                    return self[args][kwargs['collation_key']]
+                callback = Callable(partial(collation_cb, collation_key=k),
+                                    inputs=[self])
                 vdmap = self.clone(callback=callback, shared_data=False,
                                    streams=vstreams)
                 for stream in vstreams: