diff --git a/pykafka/broker.py b/pykafka/broker.py index bfcdfd15c..50b42f81d 100644 --- a/pykafka/broker.py +++ b/pykafka/broker.py @@ -17,6 +17,7 @@ limitations under the License. """ __all__ = ["Broker"] +import functools import logging import time @@ -39,6 +40,7 @@ def _check_handler(fn): """Ensures that self._req_handler is not None before calling fn""" + @functools.wraps(fn) def wrapped(self, *args, **kwargs): if self._req_handler is None: raise SocketDisconnectedError