Avoid KeyError when filtering fetchable partitions (#1344)

* Avoid KeyError when filtering fetchable partitions
dpkp · Jan 12, 2018 · 298709d · 298709d
1 parent a8bf19f
commit 298709d
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/kafka/consumer/fetcher.py b/kafka/consumer/fetcher.py
@@ -638,9 +638,9 @@ def _handle_offset_response(self, future, response):
     def _fetchable_partitions(self):
         fetchable = self._subscriptions.fetchable_partitions()
         if self._next_partition_records:
-            fetchable.remove(self._next_partition_records.topic_partition)
+            fetchable.discard(self._next_partition_records.topic_partition)
         for fetch in self._completed_fetches:
-            fetchable.remove(fetch.topic_partition)
+            fetchable.discard(fetch.topic_partition)
         return fetchable
 
     def _create_fetch_requests(self):