diff --git a/storage.py b/storage.py index 215ccf6..c933c69 100644 --- a/storage.py +++ b/storage.py @@ -264,7 +264,7 @@ class DataStorage(dict): for k in keys: if k[0] == "_": continue obj = self[k] - if isinstance(obj,(list,tuple,np.ndarray)) and all( [isinstance(v,np.ndarray) for v in obj]) and obj.ndim == 1: + if ( (isinstance(obj,np.ndarray) and obj.ndim == 1) or 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 isinstance(obj,np.ndarray): value_str = "array, size %s, type %s"% ("x".join(map(str,obj.shape)),obj.dtype)