diff --git a/storage.py b/storage.py index b3f2d06..1e632fc 100644 --- a/storage.py +++ b/storage.py @@ -239,14 +239,14 @@ class DataStorage(dict): for k in keys: if k[0] == "_": continue obj = self[k] - if isinstance(obj,np.ndarray): + if isinstance(obj,(list,tuple,np.ndarray)) and all( [isinstance(v,np.ndarray) for v in obj]): + value_str = "list of arrays, shapes " + ",".join([str(v.shape) for v in obj[:5]]) + " ..." + elif isinstance(obj,np.ndarray): value_str = "array, size %s, type %s"% ("x".join(map(str,obj.shape)),obj.dtype) elif isinstance(obj,DataStorage): value_str = str(obj)[:50] elif isinstance(obj,(str,DataStorage)): value_str = obj[:50] - elif isinstance(obj,(list,tuple)) and all( [isinstance(v,np.ndarray) for v in obj]): - value_str = "list of arrays, shapes " + ",".join([str(v.shape) for v in obj[:5]]) + " ..." elif self[k] is None: value_str = "None" else: