diff --git a/exts/omni.ext.mobility_gen/omni/ext/mobility_gen/pose_samplers.py b/exts/omni.ext.mobility_gen/omni/ext/mobility_gen/pose_samplers.py index a4b6d02..07e6c27 100644 --- a/exts/omni.ext.mobility_gen/omni/ext/mobility_gen/pose_samplers.py +++ b/exts/omni.ext.mobility_gen/omni/ext/mobility_gen/pose_samplers.py @@ -120,6 +120,8 @@ def sample_px(self, occupancy_map: OccupancyMap) -> Pose2d: # TODO: check no unoccupied coords = np.argwhere(net_mask) + if len(coords) == 0: + raise ValueError("Selected grid block contains no freespace pixels") random_index = np.random.randint(0, len(coords)) pixel = coords[random_index] pixel = Point2d(x=pixel[1], y=pixel[0])