diff --git a/tasks/interactive_idino_m2m.py b/tasks/interactive_idino_m2m.py
index f225ee5..3de3d35 100644
--- a/tasks/interactive_idino_m2m.py
+++ b/tasks/interactive_idino_m2m.py
@@ -83,7 +83,7 @@ def interactive_infer_image(model, image,all_classes,all_parts, thresh,text_size
         areas.append(area)
         mask,_=remove_small_regions(mask,int(hole_scale),mode="holes")
         mask,_=remove_small_regions(mask,int(island_scale),mode="islands")
-        mask=(mask).astype(np.float)
+        mask=(mask).astype(float)
         out_txt = texts
         visual = Visualizer(image_ori, metadata=metadata)
         color=[0.,0.,1.0]
diff --git a/tasks/interactive_predictor.py b/tasks/interactive_predictor.py
index 51732d8..e070938 100644
--- a/tasks/interactive_predictor.py
+++ b/tasks/interactive_predictor.py
@@ -85,7 +85,7 @@ def process_multi_mask(self, masks, ious, image_ori):
             areas.append(area)
             mask, _ = self.remove_small_regions(mask, int(self.hole_scale), mode="holes")
             mask, _ = self.remove_small_regions(mask, int(self.island_scale), mode="islands")
-            mask = (mask).astype(np.float)
+            mask = (mask).astype(float)
             out_txt = texts
             visual = Visualizer(image_ori, metadata=metadata)
             color = [0., 0., 1.0]