Dont raise KeyError on incomplete.remove(batch)

Author: dpkp

kafka/producer/record_accumulator.py

@@ -588,11 +588,14 @@ def __init__(self):
 
     def add(self, batch):
         with self._lock:
-            return self._incomplete.add(batch)
+            self._incomplete.add(batch)
 
     def remove(self, batch):
         with self._lock:
-            return self._incomplete.remove(batch)
+            try:
+                self._incomplete.remove(batch)
+            except KeyError:
+                pass
 
     def all(self):
         with self._lock: