improved string representation when finding list of arrays
This commit is contained in:
parent
ce3780e74a
commit
83b1de873c
|
@ -239,7 +239,7 @@ class DataStorage(dict):
|
||||||
for k in keys:
|
for k in keys:
|
||||||
if k[0] == "_": continue
|
if k[0] == "_": continue
|
||||||
obj = self[k]
|
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]]) + " ..."
|
value_str = "list of arrays, shapes " + ",".join([str(v.shape) for v in obj[:5]]) + " ..."
|
||||||
elif isinstance(obj,np.ndarray):
|
elif isinstance(obj,np.ndarray):
|
||||||
value_str = "array, size %s, type %s"% ("x".join(map(str,obj.shape)),obj.dtype)
|
value_str = "array, size %s, type %s"% ("x".join(map(str,obj.shape)),obj.dtype)
|
||||||
|
|
Loading…
Reference in New Issue