Skip to content

Batch Support for AxisAlignedBoundingBox#495

Open
zhx06 wants to merge 3 commits intomainfrom
zxiao/bbox_batch_support
Open

Batch Support for AxisAlignedBoundingBox#495
zhx06 wants to merge 3 commits intomainfrom
zxiao/bbox_batch_support

Conversation

@zhx06
Copy link
Collaborator

@zhx06 zhx06 commented Mar 23, 2026

Summary

Add batch dimension support to AxisAlignedBoundingBox

Detailed description

  • What was the reason for the change?
    • To support multi-environment batched placement, AxisAlignedBoundingBox needs to handle N bounding boxes simultaneously instead of only one.
  • What has been changed?
    • Refactored AxisAlignedBoundingBox from a dataclass to a class with (N, 3) float32 tensor storage. Returns tuples/floats for N=1, tensors for N>1.
    • Fixed float32 rounding issue in _validate_on_relations Z boundary check.
    • Added test_bounding_box.py for single-env and multi-env coverage.
  • What is the impact of this change?
    • No behavior change for existing code: backward compatible for single env (N=1).
    • Enables future multi-env batch support for the coming relation solver MR.

@zhx06 zhx06 changed the title add batch support and test cases for AxisAlignedBoundingBox Batch Support for AxisAlignedBoundingBox Mar 23, 2026
@zhx06 zhx06 force-pushed the zxiao/bbox_batch_support branch from 2e2246f to 9aa9df0 Compare March 23, 2026 20:28
@zhx06 zhx06 force-pushed the zxiao/bbox_batch_support branch from 9aa9df0 to 46af265 Compare March 23, 2026 20:40
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

1 participant