ENH: set radius to NaN when using flow or Gaussian latent prior

mj-will · Jan 17, 2025 · 3a58af7 · 3a58af7
1 parent 1a0bc57
commit 3a58af7
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/src/nessai/proposal/flowproposal/flowproposal.py b/src/nessai/proposal/flowproposal/flowproposal.py
@@ -438,7 +438,11 @@ def populate(
             min_log_q = None
 
         logger.debug(f"Populating proposal with latent radius: {r:.5}")
-        self.r = r
+        # Radius is not used for the flow or Gaussian latent priors
+        if self.latent_prior in ["flow", "gaussian"]:
+            self.r = np.nan
+        else:
+            self.r = r
 
         self.alt_dist = self.get_alt_distribution()