File tree Expand file tree Collapse file tree 1 file changed +11
-2
lines changed Expand file tree Collapse file tree 1 file changed +11
-2
lines changed Original file line number Diff line number Diff line change @@ -104,13 +104,22 @@ class Entity(OswBaseModel):
104104
105105    def  __init__ (self , ** data ):
106106        if  data .get ("label" ):
107+             if  not  isinstance (data ["label" ], list ):
108+                 raise  ValueError (
109+                     "label must be a list of Label objects" ,
110+                 )
107111            labels  =  []
108112            for  label  in  data ["label" ]:
109113                if  isinstance (label , dict ):
110-                     data [ "label" ]  =   labels .append (Label (** label ))
114+                     labels .append (Label (** label ))
111115                else :
112-                     labels .append (Label (text = label ))
116+                     # The list element should be a Label object 
117+                     labels .append (label )
113118            data ["label" ] =  labels 
119+             if  not  all (isinstance (label , Label ) for  label  in  data ["label" ]):
120+                 raise  ValueError (
121+                     "label must be a list of Label objects" ,
122+                 )
114123        if  data .get ("name" ) is  None  and  "label"  in  data :
115124            data ["name" ] =  pascal_case (data ["label" ][0 ].text )
116125        super ().__init__ (** data )
 
 
   
 
     
   
   
          
    
    
     
    
      
     
     
    You can’t perform that action at this time.
  
 
    
  
    
      
        
     
       
      
     
   
 
    
    
  
 
  
 
     
    
0 commit comments