0234315776751221ef9da5d2073e32f9f54c01bb,test/color/test_gray.py,TestBgrToGrayscale,test_gradcheck,#TestBgrToGrayscale#Any#Any#,122

Before Change

    def test_gradcheck(self, device, dtype):
        batch_size, channels, height, width = 2, 3, 4, 5
        img = torch.ones(batch_size, channels, height, width).to(device, dtype)
        img = utils.tensor_to_gradcheck_var(img)  // to var
        assert gradcheck(kornia.bgr_to_grayscale, (img,), raise_exception=True)

    def test_module(self, device, dtype):

After Change

    @pytest.mark.grad
    def test_gradcheck(self, device, dtype):
        B, C, H, W = 2, 3, 4, 4
        img = torch.ones(B, C, H, W, device=device, dtype=torch.float64, requires_grad=True)
        assert gradcheck(kornia.bgr_to_grayscale, (img,), raise_exception=True)

    @pytest.mark.jit
    def test_jit(self, device, dtype):

Instances