Update blob_shape.py

_get_exponential_shape now protects against overflow for large theta-values.
uit-cosmo · Apr 19, 2024 · b9581cc · b9581cc
1 parent 1d4b99c
commit b9581cc
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/blobmodel/blob_shape.py b/blobmodel/blob_shape.py
@@ -35,7 +35,9 @@ def _get_exponential_shape(theta: np.ndarray, **kwargs) -> np.ndarray:
     np.ndarray
         Array representing the exponential pulse shape.
     """
-    return np.exp(theta) * np.heaviside(-1.0 * theta, 1)
+    kern = np.zeros(shape=np.shape(theta))
+    kern[theta < 0] = np.exp(theta[theta < 0])
+    return kern
 
 
 def _get_lorentz_shape(theta: np.ndarray, **kwargs) -> np.ndarray: