Change some callers of add_node_before to use add_op_before

CQCL · Oct 24, 2023 · 3998b8e · 3998b8e
1 parent 7f7b799
commit 3998b8e
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 3 deletions.
diff --git a/src/builder/conditional.rs b/src/builder/conditional.rs
@@ -126,8 +126,7 @@ impl<B: AsMut<Hugr> + AsRef<Hugr>> ConditionalBuilder<B> {
         let case_node =
             // add case before any existing subsequent cases
             if let Some(&sibling_node) = self.case_nodes[case + 1..].iter().flatten().next() {
-                // TODO: Allow this to be non-pure
-                self.hugr_mut().add_node_before(sibling_node, NodeType::open_extensions(case_op))?
+                self.hugr_mut().add_op_before(sibling_node, case_op)?
             } else {
                 self.add_child_node(NodeType::open_extensions(case_op))?
             };

diff --git a/src/extension/infer.rs b/src/extension/infer.rs
@@ -1220,7 +1220,7 @@ mod test {
             }),
         )?;
 
-        let entry = hugr.add_node_before(exit, NodeType::open_extensions(dfb))?;
+        let entry = hugr.add_op_before(exit,dfb)?;
         let entry_in = hugr.add_node_with_parent(
             entry,
             NodeType::open_extensions(ops::Input { types: inputs }),