resolve issue with mdc pp plots (#187)

ML4GW · Dec 11, 2024 · 569a524 · 569a524
1 parent 8f9c721
commit 569a524
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 2 deletions.
diff --git a/amplfi/train/data/datasets/testing/mdc.py b/amplfi/train/data/datasets/testing/mdc.py
@@ -21,6 +21,8 @@ def phi_from_ra(ra: np.ndarray, gpstimes: np.ndarray) -> float:
     # calculate the relative azimuthal angle in the range [0, 2pi]
     phi = np.remainder(ra - gmsts, 2 * np.pi)
 
+    # convert to range [-pi, pi]
+    phi[phi > np.pi] -= 2 * np.pi
     return phi
 
 

diff --git a/amplfi/train/priors.py b/amplfi/train/priors.py
@@ -47,11 +47,13 @@ def cbc_prior() -> ParameterSampler:
             torch.as_tensor(0, dtype=torch.float32),
             torch.as_tensor(2 * torch.pi, dtype=torch.float32),
         ),
-        chi1=distributions.DeltaFunction(
+        chi1=Uniform(
             torch.as_tensor(0, dtype=torch.float32),
+            torch.as_tensor(0.999, dtype=torch.float32),
         ),
-        chi2=distributions.DeltaFunction(
+        chi2=Uniform(
             torch.as_tensor(0, dtype=torch.float32),
+            torch.as_tensor(0.999, dtype=torch.float32),
         ),
     )