improved string representation when finding list of arrays
This commit is contained in:
parent
2c6f7663f6
commit
5a050bb5a3
|
@ -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:
|
||||
|
|
Loading…
Reference in New Issue