don't throw when repr(imgaug) fails

tensorpack/dataflow/imgaug/base.py

@@ -9,6 +9,7 @@ import six
 from six.moves import zip
 
 from ...utils.utils import get_rng
+from ...utils.argtools import log_once
 from ..image import check_dtype
 
 __all__ = ['Augmentor', 'ImageAugmentor', 'AugmentorList']
@@ -81,23 +82,27 @@ class Augmentor(object):
         Produce something like:
         "imgaug.MyAugmentor(field1={self.field1}, field2={self.field2})"
         """
-        argspec = inspect.getargspec(self.__init__)
-        assert argspec.varargs is None, "The default __repr__ doesn't work for vaargs!"
-        assert argspec.keywords is None, "The default __repr__ doesn't work for kwargs!"
-        fields = argspec.args[1:]
-        index_field_has_default = len(fields) - (0 if argspec.defaults is None else len(argspec.defaults))
-
-        classname = type(self).__name__
-        argstr = []
-        for idx, f in enumerate(fields):
-            assert hasattr(self, f), \
-                "Attribute {} not found! The default __repr__ only works if attributes match the constructor.".format(f)
-            attr = getattr(self, f)
-            if idx >= index_field_has_default:
-                if attr is argspec.defaults[idx - index_field_has_default]:
-                    continue
-            argstr.append("{}={}".format(f, pprint.pformat(attr)))
-        return "imgaug.{}({})".format(classname, ', '.join(argstr))
+        try:
+            argspec = inspect.getargspec(self.__init__)
+            assert argspec.varargs is None, "The default __repr__ doesn't work for vaargs!"
+            assert argspec.keywords is None, "The default __repr__ doesn't work for kwargs!"
+            fields = argspec.args[1:]
+            index_field_has_default = len(fields) - (0 if argspec.defaults is None else len(argspec.defaults))
+
+            classname = type(self).__name__
+            argstr = []
+            for idx, f in enumerate(fields):
+                assert hasattr(self, f), \
+                    "Attribute {} not found! Default __repr__ only works if attributes match the constructor.".format(f)
+                attr = getattr(self, f)
+                if idx >= index_field_has_default:
+                    if attr is argspec.defaults[idx - index_field_has_default]:
+                        continue
+                argstr.append("{}={}".format(f, pprint.pformat(attr)))
+            return "imgaug.{}({})".format(classname, ', '.join(argstr))
+        except AssertionError as e:
+            log_once(e.args[0], 'warn')
+            return super(Augmentor, self).__repr__()
 
     __str__ = __repr__
 

tensorpack/dataflow/parallel_map.py

@@ -2,6 +2,7 @@
 # -*- coding: utf-8 -*-
 # File: parallel_map.py
 import numpy as np
+import time
 import ctypes
 import copy
 import threading
@@ -271,6 +272,7 @@ class MultiProcessMapDataZMQ(_ParallelMapData, _MultiProcessZMQDataFlow):
         self._iter_worker = _repeat_iter(lambda: iter(self._proc_ids))
 
         self._start_processes()
+        time.sleep(5)  # TODO temporarily work around #673
         self._fill_buffer()     # pre-fill the bufer
 
     def reset_state(self):