def __init__(self, end_point, hwm, bind=True):
        """
        Args:
            end_point (str): the ZMQ endpoint
            hwm (int): the ZMQ high-water-mark
        """
        self._end_point = end_point
        self._hwm = int(hwm)
        self._bind = bind

        def fn():
            ret = self._zmq_pull_socket.pull()
            assert len(ret) == len(self._spec)
            for qv, v in zip(ret, self._spec):
                qv.set_shape(v.shape)
            return ret
        super(ZMQInput, self).__init__(fn)

        Args:
            end_point (str): the ZMQ endpoint
            hwm (int): the ZMQ high-water-mark
        
tensorpack/input_source/input_source.py