Minor fix

tlm-adjoint · Sep 20, 2024 · 32bf6a4 · 32bf6a4
1 parent 56f6a1c
commit 32bf6a4
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/tlm_adjoint/fenics/backend_interface.py b/tlm_adjoint/fenics/backend_interface.py
@@ -235,10 +235,10 @@ def add_duplicated_comm(comm, dup_comm):
     _dup_comms.setdefault(comm_py2f, {})[dup_comm_py2f] = dup_comm
 
     def comm_finalize_callback(comm_py2f):
-        _dup_comms.pop(comm_py2f)
+        _dup_comms.pop(comm_py2f, None)
 
     def dup_comm_finalize_callback(comm_py2f, dup_comm_py2f):
-        _dup_comms.get(comm_py2f, {}).pop(dup_comm_py2f)
+        _dup_comms.get(comm_py2f, {}).pop(dup_comm_py2f, None)
 
     comm_finalize(comm, comm_finalize_callback,
                   comm_py2f)