increase numerical tolerance (#16)

mberr · Jun 21, 2022 · 8de471b · 8de471b
1 parent 28a33ed
commit 8de471b
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/torch_ppr/utils.py b/src/torch_ppr/utils.py
@@ -143,7 +143,7 @@ def validate_adjacency(adj: torch.Tensor, n: Optional[int] = None):
     else:
         # hotfix until torch.sparse.sum is implemented
         adj_sum = adj.t() @ torch.ones(adj.shape[0])
-    if not torch.allclose(adj_sum, torch.ones_like(adj_sum)):
+    if not torch.allclose(adj_sum, torch.ones_like(adj_sum), rtol=1.0e-04):
         raise ValueError(f"Invalid column sum: {adj_sum}. expected 1.0")