diff --git a/zoedepth/data/transforms.py b/zoedepth/data/transforms.py index 374416dff..0cced1912 100644 --- a/zoedepth/data/transforms.py +++ b/zoedepth/data/transforms.py @@ -372,7 +372,7 @@ def __call__(self, sample): if self.__use_mask: mask = sample["mask"] else: - mask = np.ones_like(disp, dtype=np.bool) + mask = np.ones_like(disp, dtype=bool) if np.sum(mask) == 0: return sample