bugfix in ZMQInput

tensorpack/input_source/input_source.py

@@ -425,13 +425,14 @@ class ZMQInput(TensorInput):
 
         def fn():
             ret = self._zmq_pull_socket.pull()
-            assert len(ret) == len(self._desc)
-            for qv, v in zip(ret, self._desc):
+            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)
 
     def _setup(self, input_signature):
+        super(ZMQInput, self)._setup(input_signature)
         assert len(input_signature) > 0, \
             "ZMQInput has to be used with input signature!"
 

tensorpack/tfutils/common.py

@@ -228,7 +228,7 @@ def collect_env_info():
     has_prctl = True
     try:
         import prctl
-        _ = prctl.set_pdeathsig
+        _ = prctl.set_pdeathsig  # noqa
     except Exception:
         has_prctl = False
     data.append(("python-prctl", has_prctl))