@@ -204,7 +204,7 @@ def get_mask_edges(
204204 if crop :
205205 or_vol = seg_pred | seg_gt
206206 if not or_vol .any ():
207- pred , gt = lib .zeros (seg_pred .shape , dtype = bool ), lib .zeros (seg_gt .shape , dtype = bool ) # type: ignore
207+ pred , gt = lib .zeros (seg_pred .shape , dtype = bool ), lib .zeros (seg_gt .shape , dtype = bool )
208208 return (pred , gt ) if spacing is None else (pred , gt , pred , gt ) # type: ignore
209209 channel_first = [seg_pred [None ], seg_gt [None ], or_vol [None ]]
210210 if spacing is None and not use_cucim : # cpu only erosion
@@ -275,11 +275,11 @@ def get_surface_distance(
275275 """
276276 lib : ModuleType = torch if isinstance (seg_pred , torch .Tensor ) else np
277277 if not seg_gt .any ():
278- dis = lib .inf * lib .ones_like (seg_gt , dtype = lib .float32 ) # type: ignore
278+ dis = lib .inf * lib .ones_like (seg_gt , dtype = lib .float32 )
279279 else :
280- if not lib .any (seg_pred ): # type: ignore
281- dis = lib .inf * lib .ones_like (seg_gt , dtype = lib .float32 ) # type: ignore
282- dis = dis [seg_gt ] # type: ignore
280+ if not lib .any (seg_pred ):
281+ dis = lib .inf * lib .ones_like (seg_gt , dtype = lib .float32 )
282+ dis = dis [seg_gt ]
283283 return convert_to_dst_type (dis , seg_pred , dtype = dis .dtype )[0 ]
284284 if distance_metric == "euclidean" :
285285 dis = monai_distance_transform_edt ((~ seg_gt )[None , ...], sampling = spacing )[0 ] # type: ignore
0 commit comments