From 07c09cd74f3a26897bf66cd35c69a07a16657430 Mon Sep 17 00:00:00 2001
From: Alan Lawrence <alan.lawrence@cambridgequantum.com>
Date: Thu, 30 May 2024 14:58:17 +0100
Subject: [PATCH] Fix DataflowBlock::inner_signature (fixes 2*cfg tests in
 infer.rs)

---
 hugr-core/src/ops/controlflow.rs | 1 +
 1 file changed, 1 insertion(+)

diff --git a/hugr-core/src/ops/controlflow.rs b/hugr-core/src/ops/controlflow.rs
index c09333ee3..2aba3667a 100644
--- a/hugr-core/src/ops/controlflow.rs
+++ b/hugr-core/src/ops/controlflow.rs
@@ -159,6 +159,7 @@ impl DataflowParent for DataflowBlock {
         let mut node_outputs = vec![sum_type];
         node_outputs.extend_from_slice(&self.other_outputs);
         FunctionType::new(self.inputs.clone(), TypeRow::from(node_outputs))
+            .with_extension_delta(self.extension_delta.clone())
     }
 }