diff --git a/storage.py b/storage.py index 1e632fc..81968a7 100644 --- a/storage.py +++ b/storage.py @@ -239,7 +239,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]): + if isinstance(obj,(list,tuple,np.ndarray)) and all( [isinstance(v,np.ndarray) for v in obj]) and obj.ndim == 1: 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)