From cab54baa9db422c68dd1f21f5ce6e95ce119535b Mon Sep 17 00:00:00 2001
From: Craig Roy <craig.roy@cambridgequantum.com>
Date: Tue, 12 Sep 2023 14:31:17 +0100
Subject: [PATCH 1/2] fix: Use given input extensions in `define_function`

---
 src/builder/build_traits.rs | 11 +++++++----
 1 file changed, 7 insertions(+), 4 deletions(-)

diff --git a/src/builder/build_traits.rs b/src/builder/build_traits.rs
index 09ff8c129..74e4630eb 100644
--- a/src/builder/build_traits.rs
+++ b/src/builder/build_traits.rs
@@ -86,10 +86,13 @@ pub trait Container {
         name: impl Into<String>,
         signature: Signature,
     ) -> Result<FunctionBuilder<&mut Hugr>, BuildError> {
-        let f_node = self.add_child_op(ops::FuncDefn {
-            name: name.into(),
-            signature: signature.clone().into(),
-        })?;
+        let f_node = self.add_child_node(NodeType::new(
+            ops::FuncDefn {
+                name: name.into(),
+                signature: signature.clone().into(),
+            },
+            signature.input_extensions.clone(),
+        ))?;
 
         let db = DFGBuilder::create_with_io(
             self.hugr_mut(),

From 460e1ca05c14fc84b21365e75d16ba5022ed9217 Mon Sep 17 00:00:00 2001
From: Craig Roy <craig.roy@cambridgequantum.com>
Date: Tue, 12 Sep 2023 14:37:10 +0100
Subject: [PATCH 2/2] cosmetic: Avoid `into` call

---
 src/builder/build_traits.rs | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/src/builder/build_traits.rs b/src/builder/build_traits.rs
index 74e4630eb..d3589fb39 100644
--- a/src/builder/build_traits.rs
+++ b/src/builder/build_traits.rs
@@ -89,7 +89,7 @@ pub trait Container {
         let f_node = self.add_child_node(NodeType::new(
             ops::FuncDefn {
                 name: name.into(),
-                signature: signature.clone().into(),
+                signature: signature.signature.clone(),
             },
             signature.input_extensions.clone(),
         ))?;