Expose method to list communities' edges (#18)

* Expose method to list communities' edges

* changelog

CHANGELOG.md

@@ -3,3 +3,4 @@
 ### Features
 - Added exporting the graph to `scipy.csr_matrix` ([#16](https://github.com/AleksanderWWW/abcd-graph/pull/16))
 - Added a function to set the random seed ([#17](https://github.com/AleksanderWWW/abcd-graph/pull/17))
+- Expose method to list communities' edges ([#18](https://github.com/AleksanderWWW/abcd-graph/pull/18))

src/abcd_graph/abcd.py

@@ -227,3 +227,10 @@ def build(self, model: Optional[Model] = None) -> "Graph":
         self._graph.rewire_graph()
 
         return self
+
+    @property
+    @require_graph_built
+    def vertex_partition(self) -> dict[int, list[int]]:
+        assert self._graph is not None
+
+        return {i: community.vertices for i, community in enumerate(self._graph.communities)}