Fix to noise generation for older numpy versions

pints-team · Oct 31, 2018 · f248c48 · f248c48
1 parent 6fb51de
commit f248c48
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/pints/noise.py b/pints/noise.py
@@ -75,7 +75,11 @@ def ar1(rho, sigma, n):
         return np.array([])
 
     # Generate noise
-    v = np.random.normal(0, sigma * np.sqrt(1 - rho**2), n)
+    s = sigma * np.sqrt(1 - rho**2)
+    if s == 0:
+        v = np.zeros(n)
+    else:
+        v = np.random.normal(0, s, n)
     v[0] = np.random.rand()
     for t in range(1, n):
         v[t] += rho * v[t - 1]