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:
 | 
					    for k in keys:
 | 
				
			||||||
      if k[0] == "_": continue
 | 
					      if k[0] == "_": continue
 | 
				
			||||||
      obj = self[k]
 | 
					      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)
 | 
					        value_str = "array, size %s, type %s"% ("x".join(map(str,obj.shape)),obj.dtype)
 | 
				
			||||||
      elif isinstance(obj,DataStorage):
 | 
					      elif isinstance(obj,DataStorage):
 | 
				
			||||||
        value_str = str(obj)[:50]
 | 
					        value_str = str(obj)[:50]
 | 
				
			||||||
      elif isinstance(obj,(str,DataStorage)):
 | 
					      elif isinstance(obj,(str,DataStorage)):
 | 
				
			||||||
        value_str = obj[:50] 
 | 
					        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:
 | 
					      elif self[k] is None:
 | 
				
			||||||
        value_str = "None"
 | 
					        value_str = "None"
 | 
				
			||||||
      else:
 | 
					      else:
 | 
				
			||||||
| 
						 | 
					
 | 
				
			||||||
		Loading…
	
		Reference in New Issue