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