From d2646b3c8a49c39fdfdff3def6be447c185e2d56 Mon Sep 17 00:00:00 2001
From: "Christian Y. Brenninkmeijer"
 <christian.brenninkmeijer@manchester.ac.uk>
Date: Tue, 9 Jan 2024 07:28:54 +0000
Subject: [PATCH] check for None

---
 mcmc/mcmc_coordinator_vertex.py | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/mcmc/mcmc_coordinator_vertex.py b/mcmc/mcmc_coordinator_vertex.py
index c2131bf..6e42ba0 100644
--- a/mcmc/mcmc_coordinator_vertex.py
+++ b/mcmc/mcmc_coordinator_vertex.py
@@ -226,7 +226,8 @@ def generate_data_specification(
             if self._is_receiver_placement(mcmc_placement):
                 key = routing_info.get_first_key_from_pre_vertex(
                     vertex, self._acknowledge_partition_name)
-                keys.append(key)
+                if key is not None:
+                    keys.append(key)
         keys.sort()
 
         # Write the data size in words