make the bounding box to be float32 data type

facebookresearch · Oct 27, 2023 · 5d80be4 · 5d80be4
1 parent 8d7a696
commit 5d80be4
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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)