@@ -141,10 +141,7 @@ def minmax(lst, old):
141141 # soma: modified when ctng properly assigns soma obj to segments
142142 if soma_objects :
143143 for item , seg in soma_objects .items ():
144- if seg in final_seg_dict :
145- final_seg_dict [seg ].append (item )
146- else :
147- final_seg_dict [seg ] = [item ]
144+ final_seg_dict .setdefault (seg , []).append (item )
148145
149146 # assign join objects
150147 for jg in join_groups :
@@ -154,18 +151,12 @@ def minmax(lst, old):
154151 if (not (isinstance (item , Cone ) or isinstance (item , Cylinder ))) or (
155152 item in join_objects
156153 ):
157- if seg in final_seg_dict .keys ():
158- final_seg_dict [seg ].append (item )
159- else :
160- final_seg_dict [seg ] = [item ]
154+ final_seg_dict .setdefault (seg , []).append (item )
161155
162156 # complete final segment dictionary
163157 for cone in cones :
164158 seg = segment_dict [(cone ._x0 , cone ._y0 , cone ._z0 , cone ._x1 , cone ._y1 , cone ._z1 )]
165- if seg in final_seg_dict .keys ():
166- final_seg_dict [seg ].append (cone )
167- else :
168- final_seg_dict [seg ] = [cone ]
159+ final_seg_dict .setdefault (seg , []).append (cone )
169160
170161 # voxelize all the objects and assign voxels
171162 # output dictionaries of internal and surface voxels
0 commit comments