add wrapper

Qiskit · Dec 8, 2024 · eed8af1 · eed8af1
1 parent 66ff464
commit eed8af1
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 1 deletion.
diff --git 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
@@ -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))?;