simple_replace.rs: use HugrMut::remove_node, includes clearing op_typ…

…es (#242)
CQCL · Jul 4, 2023 · 17c24e3 · 17c24e3
1 parent 29e869d
commit 17c24e3
Showing 1 changed file with 2 additions and 3 deletions.
diff --git a/src/hugr/rewrite/simple_replace.rs b/src/hugr/rewrite/simple_replace.rs
@@ -3,7 +3,7 @@
 use std::collections::{HashMap, HashSet};
 
 use itertools::Itertools;
-use portgraph::{LinkMut, LinkView, MultiMut, NodeIndex, PortMut, PortView};
+use portgraph::{LinkMut, LinkView, MultiMut, NodeIndex, PortView};
 
 use crate::hugr::{HugrMut, HugrView};
 use crate::{
@@ -227,8 +227,7 @@ impl Rewrite for SimpleReplacement {
  }
  // 3.5. Remove all nodes in self.removal and edges between them.
  for node in &self.removal {
- h.graph.remove_node(node.index);
- h.hierarchy.remove(node.index);
+ h.remove_node(*node).unwrap();
  }
  Ok(())
  }