doctests

pyg-team · Aug 13, 2018 · 8313ae1 · 8313ae1
1 parent 0f284ac
commit 8313ae1
Show file tree

Hide file tree

Showing 7 changed files with 21 additions and 21 deletions.
diff --git a/torch_geometric/transforms/cartesian.py b/torch_geometric/transforms/cartesian.py
@@ -34,10 +34,10 @@ class Cartesian(object):
 
     .. testoutput::
 
-        tensor([[ 0.7500,  0.5000],
-                [ 0.2500,  0.5000],
-                [ 1.0000,  0.5000],
-                [ 0.0000,  0.5000]])
+        tensor([[0.7500, 0.5000],
+                [0.2500, 0.5000],
+                [1.0000, 0.5000],
+                [0.0000, 0.5000]])
     """
 
     def __init__(self, cat=True):

diff --git a/torch_geometric/transforms/compose.py b/torch_geometric/transforms/compose.py
@@ -25,8 +25,8 @@ class Compose(object):
 
     .. testoutput::
 
-        tensor([[ 0.7500,  0.5000,  1.0000],
-                [ 1.0000,  0.5000,  1.0000]])
+        tensor([[0.7500, 0.5000, 1.0000],
+                [1.0000, 0.5000, 1.0000]])
     """
 
     def __init__(self, transforms):

diff --git a/torch_geometric/transforms/local_cartesian.py b/torch_geometric/transforms/local_cartesian.py
@@ -35,10 +35,10 @@ class LocalCartesian(object):
 
     .. testoutput::
 
-        tensor([[ 1.0000,  0.5000],
-                [ 0.2500,  0.5000],
-                [ 1.0000,  0.5000],
-                [ 0.0000,  0.5000]])
+        tensor([[1.0000, 0.5000],
+                [0.2500, 0.5000],
+                [1.0000, 0.5000],
+                [0.0000, 0.5000]])
     """
 
     def __init__(self, cat=True):

diff --git a/torch_geometric/transforms/polar.py b/torch_geometric/transforms/polar.py
@@ -31,10 +31,10 @@ class Polar(object):
 
     .. testoutput::
 
-        tensor([[ 0.5000,  0.0000],
-                [ 0.5000,  0.5000],
-                [ 1.0000,  0.2500],
-                [ 1.0000,  0.7500]])
+        tensor([[0.5000, 0.0000],
+                [0.5000, 0.5000],
+                [1.0000, 0.2500],
+                [1.0000, 0.7500]])
     """
 
     def __init__(self, cat=True):

diff --git a/torch_geometric/transforms/spherical.py b/torch_geometric/transforms/spherical.py
@@ -31,8 +31,8 @@ class Spherical(object):
 
     .. testoutput::
 
-        tensor([[ 1.0000,  0.2500,  0.0000],
-                [ 1.0000,  0.7500,  1.0000]])
+        tensor([[1.0000, 0.2500, 0.0000],
+                [1.0000, 0.7500, 1.0000]])
     """
 
     def __init__(self, cat=True):

diff --git a/torch_geometric/transforms/target_indegree.py b/torch_geometric/transforms/target_indegree.py
@@ -34,10 +34,10 @@ class TargetIndegree(object):
 
     .. testoutput::
 
-        tensor([[ 1.0000],
-                [ 0.5000],
-                [ 0.5000],
-                [ 1.0000]])
+        tensor([[1.0000],
+                [0.5000],
+                [0.5000],
+                [1.0000]])
     """
 
     def __init__(self, cat=True):

diff --git a/torch_geometric/utils/degree.py b/torch_geometric/utils/degree.py
@@ -30,7 +30,7 @@ def degree(index, num_nodes=None, dtype=None, device=None):
 
     .. testoutput::
 
-       tensor([ 3.,  1.,  1.])
+       tensor([3., 1., 1.])
     """
     num_nodes = maybe_num_nodes(index, num_nodes)
     out = torch.zeros((num_nodes), dtype=dtype, device=device)