ErdosRenyi: 0 and 1 are valid probabilities

epfl-lts2 · Sep 19, 2017 · a3bf547 · a3bf547
1 parent 3381687
commit a3bf547
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/pygsp/graphs/erdosrenyi.py b/pygsp/graphs/erdosrenyi.py
@@ -36,7 +36,7 @@ def __init__(self, N=100, p=0.1, connected=False, directed=False,
                  max_iter=10, **kwargs):
         self.p = p
 
-        if not 0 < p < 1:
+        if not 0 <= p <= 1:
             raise ValueError('Probability p should be in [0, 1].')
 
         M = int(N * (N-1) if directed else N * (N-1) / 2)