Merge pull request #74 from zipengwang98/prior_fix

Update prior.py
kazewong · Mar 11, 2024 · 1589dbb · 1589dbb
2 parents 031920d + 4ab1b63
commit 1589dbb
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/jimgw/prior.py b/src/jimgw/prior.py
@@ -270,8 +270,8 @@ def log_prob(self, x: dict[str, Float]) -> Float:
             | (mag < 0)
             | (phi > 2 * jnp.pi)
             | (phi < 0)
-            | (theta > 1)
-            | (theta < -1),
+            | (theta > jnp.pi)
+            | (theta < 0),
             jnp.zeros_like(0) - jnp.inf,
             jnp.log(mag**2 * jnp.sin(x[self.naming[0]])),
         )