From eed8af1cba4d5cd89335bc3e1fe7d61bf170933c Mon Sep 17 00:00:00 2001
From: maleicacid <4982384+kazuki0824@users.noreply.github.com>
Date: Mon, 9 Dec 2024 02:07:32 +0900
Subject: [PATCH] add wrapper

---
 src/connectivity/mod.rs | 2 +-
 src/lib.rs              | 1 +
 2 files changed, 2 insertions(+), 1 deletion(-)

diff --git a/src/connectivity/mod.rs b/src/connectivity/mod.rs
index 6a2337897..c0ae69158 100644
--- a/src/connectivity/mod.rs
+++ b/src/connectivity/mod.rs
@@ -158,7 +158,7 @@ pub fn condensation_inner<'a, N, E, Ty, Ix>(
 }
 
 #[pyfunction]
-#[pyo3(text_signature = "(graph, /, sccs=None)", signature=(graph, /, sccs=None))]
+#[pyo3(text_signature = "(graph, /, sccs=None)", signature=(graph, sccs=None))]
 pub fn condensation(py: Python, graph: &digraph::PyDiGraph, sccs: Option<Vec<Vec<usize>>>)
     -> digraph::PyDiGraph {
     let g = graph.graph.clone();
diff --git a/src/lib.rs b/src/lib.rs
index 4ee4189a7..21661b4e8 100644
--- a/src/lib.rs
+++ b/src/lib.rs
@@ -569,6 +569,7 @@ fn rustworkx(py: Python<'_>, m: &Bound<PyModule>) -> PyResult<()> {
     m.add_wrapped(wrap_pyfunction!(cycle_basis))?;
     m.add_wrapped(wrap_pyfunction!(simple_cycles))?;
     m.add_wrapped(wrap_pyfunction!(strongly_connected_components))?;
+    m.add_wrapped(wrap_pyfunction!(condensation))?;
     m.add_wrapped(wrap_pyfunction!(digraph_dfs_edges))?;
     m.add_wrapped(wrap_pyfunction!(graph_dfs_edges))?;
     m.add_wrapped(wrap_pyfunction!(digraph_find_cycle))?;