export PullForwardError to python

CQCL · Sep 25, 2023 · 1187707 · 1187707
1 parent 6a7eaae
commit 1187707
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 0 deletions.
diff --git a/pyrs/src/lib.rs b/pyrs/src/lib.rs
@@ -71,6 +71,10 @@ fn add_pass_module(py: Python, parent: &PyModule) -> PyResult<()> {
     let m = PyModule::new(py, "passes")?;
     m.add_function(wrap_pyfunction!(greedy_depth_reduce, m)?)?;
     m.add_class::<tket2::T2Op>()?;
+    m.add(
+        "PullForwardError",
+        py.get_type::<tket2::passes::PyPullForwardError>(),
+    )?;
     parent.add_submodule(m)?;
     Ok(())
 }
diff --git a/src/passes.rs b/src/passes.rs
@@ -6,6 +6,8 @@
 // pub mod squash;
 mod commutation;
 pub use commutation::apply_greedy_commutation;
+#[cfg(feature = "pyo3")]
+pub use commutation::PyPullForwardError;
 
 // use rayon::prelude::*;