Improve code readability

cvxgrp · Oct 29, 2024 · 5a988d1 · 5a988d1
1 parent 2eb509e
commit 5a988d1
Showing 1 changed file with 4 additions and 5 deletions.
diff --git a/src/spcqe/extrapolate_asymptotic.py b/src/spcqe/extrapolate_asymptotic.py
@@ -59,11 +59,10 @@ def safe_exp(exp_input):
     return exp_output
 
 def safe_log(log_input):
-    return np.where(
-        log_input <= 0,
-        -np.inf,  # Return -inf for invalid log input
-        np.log(log_input)
-    )
+    log_output = np.empty_like(log_input)
+    log_output[log_input <= 0] = -np.inf
+    log_output[log_input > 0] = np.log(log_input[log_input > 0])
+    return log_output
 
 
 def plot_pdf(ax, transf, label):