Update doctest

torch_scatter/add.py

@@ -66,8 +66,8 @@ def scatter_add(src, index, dim=-1, out=None, dim_size=None, fill_value=0):
 
     .. testoutput::
 
-       tensor([[0, 0, 4, 3, 3, 0],
-               [2, 4, 4, 0, 0, 0]])
+       tensor([[0., 0., 4., 3., 3., 0.],
+               [2., 4., 4., 0., 0., 0.]])
     """
     src, out, index, dim = gen(src, index, dim, out, dim_size, fill_value)
     return out.scatter_add_(dim, index, src)

torch_scatter/max.py

@@ -93,8 +93,8 @@ def scatter_max(src, index, dim=-1, out=None, dim_size=None, fill_value=None):
 
     .. testoutput::
 
-       tensor([[0, 0, 4, 3, 2, 0],
-               [2, 4, 3, 0, 0, 0]])
+       tensor([[0., 0., 4., 3., 2., 0.],
+               [2., 4., 3., 0., 0., 0.]])
        tensor([[-1, -1,  3,  4,  0,  1],
                [ 1,  4,  3, -1, -1, -1]])
     """

torch_scatter/min.py

@@ -95,8 +95,8 @@ def scatter_min(src, index, dim=-1, out=None, dim_size=None, fill_value=None):
 
     .. testoutput::
 
-       tensor([[ 0,  0, -4, -3, -2,  0],
-               [-2, -4, -3,  0,  0,  0]])
+       tensor([[ 0.,  0., -4., -3., -2.,  0.],
+               [-2., -4., -3.,  0.,  0.,  0.]])
        tensor([[-1, -1,  3,  4,  0,  1],
                [ 1,  4,  3, -1, -1, -1]])
     """

torch_scatter/mul.py

@@ -84,8 +84,8 @@ def scatter_mul(src, index, dim=-1, out=None, dim_size=None, fill_value=1):
 
     .. testoutput::
 
-       tensor([[1, 1, 4, 3, 6, 0],
-               [6, 4, 8, 1, 1, 1]])
+       tensor([[1., 1., 4., 3., 6., 0.],
+               [6., 4., 8., 1., 1., 1.]])
     """
     src, out, index, dim = gen(src, index, dim, out, dim_size, fill_value)
     if src.size(dim) == 0:  # pragma: no cover

torch_scatter/sub.py

@@ -58,7 +58,7 @@ def scatter_sub(src, index, dim=-1, out=None, dim_size=None, fill_value=0):
 
     .. testoutput::
 
-       tensor([[ 0,  0, -4, -3, -3,  0],
-               [-2, -4, -4,  0,  0,  0]])
+       tensor([[ 0.,  0., -4., -3., -3.,  0.],
+               [-2., -4., -4.,  0.,  0.,  0.]])
     """
     return scatter_add(src.neg(), index, dim, out, dim_size, fill_value)