diff --git a/habitat-lab/habitat/tasks/rearrange/rearrange_sensors.py b/habitat-lab/habitat/tasks/rearrange/rearrange_sensors.py index 041e584458..991f37ff81 100644 --- a/habitat-lab/habitat/tasks/rearrange/rearrange_sensors.py +++ b/habitat-lab/habitat/tasks/rearrange/rearrange_sensors.py @@ -1300,4 +1300,4 @@ def get_observation(self, observations, episode, task, *args, **kwargs): rmin, rmax, cmin, cmax = self._get_bbox(tgt_mask) bbox[rmin:rmax, cmin:cmax] = 1.0 - return bbox + return np.float32(bbox)