@@ -521,7 +521,7 @@ def correct_crop_centers(
521521 for c , v_s , v_e in zip (centers , valid_start , valid_end ):
522522 center_i = min (max (c , v_s ), v_e - 1 )
523523 valid_centers .append (int (center_i ))
524- return ensure_tuple (valid_centers ) # type: ignore
524+ return ensure_tuple (valid_centers )
525525
526526
527527def generate_pos_neg_label_crop_centers (
@@ -579,7 +579,7 @@ def generate_pos_neg_label_crop_centers(
579579 # shift center to range of valid centers
580580 centers .append (correct_crop_centers (center , spatial_size , label_spatial_shape , allow_smaller ))
581581
582- return ensure_tuple (centers ) # type: ignore
582+ return ensure_tuple (centers )
583583
584584
585585def generate_label_classes_crop_centers (
@@ -639,7 +639,7 @@ def generate_label_classes_crop_centers(
639639 # shift center to range of valid centers
640640 centers .append (correct_crop_centers (center , spatial_size , label_spatial_shape , allow_smaller ))
641641
642- return ensure_tuple (centers ) # type: ignore
642+ return ensure_tuple (centers )
643643
644644
645645def create_grid (
@@ -2218,7 +2218,7 @@ def distance_transform_edt(
22182218 if not r_vals :
22192219 return None
22202220 device = img .device if isinstance (img , torch .Tensor ) else None
2221- return convert_data_type (r_vals [0 ] if len (r_vals ) == 1 else r_vals , output_type = type (img ), device = device )[0 ] # type: ignore
2221+ return convert_data_type (r_vals [0 ] if len (r_vals ) == 1 else r_vals , output_type = type (img ), device = device )[0 ]
22222222
22232223
22242224if __name__ == "__main__" :
0 commit comments