Add pub type DivergeCallback

CQCL · Jan 29, 2025 · 60a5823 · 60a5823
1 parent cb7bf75
commit 60a5823
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/hugr-passes/src/dead_code.rs b/hugr-passes/src/dead_code.rs
@@ -13,7 +13,7 @@ use crate::validation::{ValidatePassError, ValidationLevel};
 #[derive(Clone, Default)]
 pub struct DeadCodeElimPass {
     entry_points: Vec<Node>,
-    diverge_callback: Option<Arc<dyn Fn(&Hugr, Node) -> NodeDivergence>>,
+    diverge_callback: Option<Arc<DivergeCallback>>,
     validation: ValidationLevel,
 }
 
@@ -38,6 +38,8 @@ impl Debug for DeadCodeElimPass {
     }
 }
 
+pub type DivergeCallback = dyn Fn(&Hugr, Node) -> NodeDivergence;
+
 pub enum NodeDivergence {
     #[allow(unused)]
     MustKeep,
@@ -67,7 +69,7 @@ impl DeadCodeElimPass {
     /// [Call]: hugr_core::ops::Call
     /// [CFG]: hugr_core::ops::CFG
     /// [TailLoop]: hugr_core::ops::TailLoop
-    pub fn set_diverge_callback(mut self, cb: Arc<dyn Fn(&Hugr, Node) -> NodeDivergence>) -> Self {
+    pub fn set_diverge_callback(mut self, cb: Arc<DivergeCallback>) -> Self {
         self.diverge_callback = Some(cb);
         self
     }