Merge branch 'dev' into fix/issue-8775-compute-shape-offset-pytorch29

Signed-off-by: Eric Kerfoot <17726042+ericspod@users.noreply.github.com>

Author: ericspod

.github/workflows/pythonapp.yml

@@ -82,7 +82,7 @@ jobs:
         find  /opt/hostedtoolcache/* -maxdepth 0 ! -name 'Python' -exec rm -rf {} \;
     - name: Install the dependencies
       run: |
-        python -m pip install --user --upgrade pip wheel
+        python -m pip install --user --upgrade pip wheel pybind11
         python -m pip install torch==2.5.1 torchvision==0.20.1
         cat "requirements-dev.txt"
         python -m pip install --no-build-isolation -r requirements-dev.txt

requirements-dev.txt

@@ -18,6 +18,7 @@ black==25.1.0
 isort>=5.1, <6, !=6.0.0
 ruff>=0.14.11,<0.15
 pytype>=2020.6.1, <=2024.4.11; platform_system != "Windows"
+pybind11
 types-setuptools
 mypy>=1.5.0, <1.12.0
 ninja

tests/data/utils/test_compute_shape_offset.py

@@ -21,6 +21,26 @@
 
 class TestComputeShapeOffset(unittest.TestCase):
     """Unit tests for :func:`monai.data.utils.compute_shape_offset`."""
+    
+    def test_pytorch_size_input(self):
+        """Validate `torch.Size` input produces expected shape and offset.
+
+        Returns:
+            None.
+
+        Raises:
+            AssertionError: If computed shape/offset are not as expected.
+        """
+        # 1. Create a PyTorch Size object (which triggered the original bug)
+        spatial_shape = torch.Size([10, 10, 10])
+        in_affine = np.eye(4)
+        out_affine = np.eye(4)
+
+        # 2. Feed it into the function
+        shape, offset = compute_shape_offset(spatial_shape, in_affine, out_affine)
+
+        # 3. Prove it successfully processed the shape by checking its length
+        self.assertEqual(len(shape), 3)
 
     def setUp(self):
         """Set up a 4x4 identity affine used across all test cases."""