feat: Default more builder methods to open extensions

CQCL · Sep 4, 2023 · c66b266 · c66b266
1 parent c6bfb1e
commit c66b266
Show file tree

Hide file tree

Showing 4 changed files with 7 additions and 9 deletions.
diff --git a/src/builder/build_traits.rs b/src/builder/build_traits.rs
@@ -322,8 +322,7 @@ pub trait Dataflow: Container {
 
         let (cfg_node, _) = add_node_with_wires(
             self,
-            // TODO: Make input extensions a parameter
-            NodeType::pure(ops::CFG {
+            NodeType::open_extensions(ops::CFG {
                 inputs: inputs.clone(),
                 outputs: output_types.clone(),
             }),
@@ -606,8 +605,7 @@ fn add_op_with_wires<T: Dataflow + ?Sized>(
     optype: impl Into<OpType>,
     inputs: Vec<Wire>,
 ) -> Result<(Node, usize), BuildError> {
-    // TODO: Make this NodeType::open_extensions
-    add_node_with_wires(data_builder, NodeType::pure(optype), inputs)
+    add_node_with_wires(data_builder, NodeType::open_extensions(optype), inputs)
 }
 
 fn add_node_with_wires<T: Dataflow + ?Sized>(

diff --git a/src/builder/cfg.rs b/src/builder/cfg.rs
@@ -63,7 +63,7 @@ impl CFGBuilder<Hugr> {
         };
 
         // TODO: Allow input extensions to be specified
-        let base = Hugr::new(NodeType::pure(cfg_op));
+        let base = Hugr::new(NodeType::open_extensions(cfg_op));
         let cfg_node = base.root();
         CFGBuilder::create(base, cfg_node, input, output)
     }
@@ -280,7 +280,7 @@ impl BlockBuilder<Hugr> {
         };
 
         // TODO: Allow input extensions to be specified
-        let base = Hugr::new(NodeType::pure(op));
+        let base = Hugr::new(NodeType::open_extensions(op));
         let root = base.root();
         Self::create(base, root, predicate_variants, other_outputs, inputs)
     }

diff --git a/src/builder/conditional.rs b/src/builder/conditional.rs
@@ -178,7 +178,7 @@ impl ConditionalBuilder<Hugr> {
             extension_delta,
         };
         // TODO: Allow input extensions to be specified
-        let base = Hugr::new(NodeType::pure(op));
+        let base = Hugr::new(NodeType::open_extensions(op));
         let conditional_node = base.root();
 
         Ok(ConditionalBuilder {
@@ -197,7 +197,7 @@ impl CaseBuilder<Hugr> {
             signature: signature.clone(),
         };
         // TODO: Allow input extensions to be specified
-        let base = Hugr::new(NodeType::pure(op));
+        let base = Hugr::new(NodeType::open_extensions(op));
         let root = base.root();
         let dfg_builder = DFGBuilder::create_with_io(base, root, signature, None)?;
 

diff --git a/src/builder/tail_loop.rs b/src/builder/tail_loop.rs
@@ -82,7 +82,7 @@ impl TailLoopBuilder<Hugr> {
             rest: inputs_outputs.into(),
         };
         // TODO: Allow input extensions to be specified
-        let base = Hugr::new(NodeType::pure(tail_loop.clone()));
+        let base = Hugr::new(NodeType::open_extensions(tail_loop.clone()));
         let root = base.root();
         Self::create_with_io(base, root, &tail_loop)
     }