feat: Open up extensions in some builder methods

CQCL · Aug 31, 2023 · 60f4292 · 60f4292
1 parent ba306fb
commit 60f4292
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 5 deletions.
diff --git a/src/builder/build_traits.rs b/src/builder/build_traits.rs
@@ -282,8 +282,7 @@ pub trait Dataflow: Container {
             signature: signature.clone(),
         };
         let nodetype = match &input_extensions {
-            // TODO: Make this NodeType::open_extensions
-            None => NodeType::pure(op),
+            None => NodeType::open_extensions(op),
             Some(rs) => NodeType::new(op, rs.clone()),
         };
         let (dfg_n, _) = add_node_with_wires(self, nodetype, input_wires.into_iter().collect())?;

diff --git a/src/builder/dataflow.rs b/src/builder/dataflow.rs
@@ -52,16 +52,15 @@ impl<T: AsMut<Hugr> + AsRef<Hugr>> DFGBuilder<T> {
         base.as_mut().add_node_with_parent(
             parent,
             match &input_extensions {
-                // TODO: Make this NodeType::open_extensions
-                None => NodeType::pure(input),
+                None => NodeType::open_extensions(input),
                 Some(rs) => NodeType::new(input, rs.clone()),
             },
         )?;
         base.as_mut().add_node_with_parent(
             parent,
             match input_extensions.map(|inp| inp.union(&signature.extension_reqs)) {
                 // TODO: Make this NodeType::open_extensions
-                None => NodeType::new(output, signature.extension_reqs),
+                None => NodeType::open_extensions(output),
                 Some(rs) => NodeType::new(output, rs),
             },
         )?;