transpose bounds if necessary, avoid confusion

kazewong · Dec 6, 2023 · ed223ef · ed223ef
1 parent 282f409
commit ed223ef
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/src/jimgw/likelihood.py b/src/jimgw/likelihood.py
@@ -179,6 +179,10 @@ def __init__(
         self.freq_grid_low = freq_grid[:-1]
 
         print("Finding reference parameters..")
+
+        # Check if shape of bounds is correct, otherwise, transpose bounds
+        if jnp.shape(bounds)[0] != 2:
+            bounds = bounds.T
 
         self.ref_params = self.maximize_likelihood(
             bounds=bounds, prior=prior, popsize=popsize, n_loops=n_loops