From 4ab1b63b03cdb6f8dd71660d43ca2a8890cf6edd Mon Sep 17 00:00:00 2001
From: Zipeng Wang <scientistwang9877@gmail.com>
Date: Mon, 11 Mar 2024 11:03:19 -0400
Subject: [PATCH] Update prior.py

---
 src/jimgw/prior.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/src/jimgw/prior.py b/src/jimgw/prior.py
index f8aab8e5..32107e1e 100644
--- 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]])),
         )