diff options
| -rw-r--r-- | kafka/producer/kafka.py | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/kafka/producer/kafka.py b/kafka/producer/kafka.py index e1200f4..63cff73 100644 --- a/kafka/producer/kafka.py +++ b/kafka/producer/kafka.py @@ -314,7 +314,7 @@ class KafkaProducer(object): return wrapper def _unregister_cleanup(self): - if getattr(self, '_cleanup'): + if getattr(self, '_cleanup', None): if hasattr(atexit, 'unregister'): atexit.unregister(self._cleanup) # pylint: disable=no-member |