plot_patterns_hypergraph return axes

INM-6 · Nov 16, 2023 · 45904ea · 45904ea
1 parent 1fd9481
commit 45904ea
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/viziphant/patterns_src/view.py b/viziphant/patterns_src/view.py
@@ -11,6 +11,7 @@
 from holoviews import opts
 from holoviews.streams import Pipe
 import numpy as np
+import matplotlib.pyplot as plt
 
 from viziphant.patterns_src.hypergraph import Hypergraph
 
@@ -230,7 +231,8 @@ def show(self,
         # Set size of the plot to a square to avoid distortions
         self.plot = plot.redim.range(x=(-1, 11), y=(-1, 11))
 
-        return hv.render(plot, backend="matplotlib")
+        axes = hv.render(plot, backend="matplotlib").axes[0]
+        return axes
 
     def draw_hyperedges(self,
                         subset_style=VisualizationStyle.COLOR,