08143b8126d203ece62227f3b4bf157eeff5e58f,test/test_imgwarp.py,TestRemap,test_jit,#TestRemap#,366

Before Change

        grid += 1.  // apply some shift

        input = (img, grid[..., 0], grid[..., 1],)
        remap_traced = torch.jit.trace(tgm.remap, input)

        assert_allclose(tgm.remap(*input), remap_traced(*input))

After Change

        input = (img, grid[..., 0], grid[..., 1],)
        actual = op_script(*input)
        expected = tgm.remap(*input)
        assert_allclose(actual, expected)

    def test_jit_trace(self):
        @torch.jit.script
        def op_script(input, map1, map2):

Instances