a828315185a9dc8b21ec8e5dbead9044caf0d3a2,test/geometry/test_linalg.py,TestComposeTransforms,test_gradcheck,#TestComposeTransforms#Any#Any#,230

Before Change

    @pytest.mark.parametrize("batch_size", [1, 2, 5])
    def test_gradcheck(self, device, batch_size):
        trans_01 = identity_matrix(batch_size).to(device)
        trans_12 = identity_matrix(batch_size).to(device)

        trans_01 = utils.tensor_to_gradcheck_var(trans_01)  // to var
        trans_12 = utils.tensor_to_gradcheck_var(trans_12)  // to var

After Change

        assert gradcheck(kornia.transform_boxes, (trans_mat, boxes), raise_exception=True)

    @pytest.mark.skip(reason="turn off all jit for a while")
    def test_jit(self, device, dtype):
        @torch.jit.script
        def op_script(transform, boxes):
            return kornia.transform_boxes(transform, boxes)

        boxes = torch.tensor([139.2640, 103.0150, 258.0480, 307.5075], device=device, dtype=dtype)

Instances