diff --git a/viziphant/patterns.py b/viziphant/patterns.py index 26a2afc..25c176d 100644 --- a/viziphant/patterns.py +++ b/viziphant/patterns.py @@ -399,6 +399,7 @@ def plot_patterns(spiketrains, patterns, circle_sizes=(3, 50, 70), axes.yaxis.set_label_coords(-0.01, 0.5) return axes +# TODO: add a parameter node_size def plot_patterns_hypergraph(patterns, num_neurons=None): """ Hypergraph visualization of spike patterns. @@ -461,7 +462,7 @@ def plot_patterns_hypergraph(patterns, num_neurons=None): bst.rescale('ms') patterns = cell_assembly_detection(bst, max_lag=2) - fig = viziphant.patterns.plot_patterns_hypergraph(patterns) + viziphant.patterns.plot_patterns_hypergraph(patterns) plt.show() """ diff --git a/viziphant/patterns_src/view.py b/viziphant/patterns_src/view.py index ff50c91..a440d7a 100644 --- a/viziphant/patterns_src/view.py +++ b/viziphant/patterns_src/view.py @@ -53,8 +53,8 @@ def __init__(self, hypergraphs, title=None): # Which color of the color map to use next self.current_color = 1 - # Size of the vertices - self.node_radius = 0.2 + # Size of the vertices TODO: add as parameter + self.node_radius = .2 # Selected title of the figure self.title = title @@ -225,8 +225,8 @@ def show(self, plot = self.dynamic_map * self.dynamic_map_edges # Set size of the plot to a square to avoid distortions self.plot = plot.redim.range(x=(-1, 11), y=(-1, 11)) - - axes = hv.render(plot, backend="matplotlib").axes[0] + # TODO: how to get axes? currently figure + axes = hv.render(plot, backend="matplotlib") return axes def draw_hyperedges(self,