.loc to avoid warnings (#255)

* use to_numpy() on boolean mask to avoid warning * remove print statements * Converted .to_numpy to .log * Removed extraneous brackets --------- Co-authored-by: Will Handley <[email protected]>
handley-lab · Feb 7, 2023 · 67b9a96 · 67b9a96
1 parent 0618a8f
commit 67b9a96
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/anesthetic/examples/perfect_ns.py b/anesthetic/examples/perfect_ns.py
@@ -53,8 +53,8 @@ def random_sphere(n):
  r = (points**2).sum(axis=-1, keepdims=True)**0.5
 
  samples = merge_nested_samples(samples)
- logLend = samples[samples.nlive >= nlive].logL.max()
- return samples[samples.logL_birth < logLend].recompute()
+ logLend = samples.loc[samples.nlive >= nlive].logL.max()
+ return samples.loc[samples.logL_birth < logLend].recompute()
 
 
 def correlated_gaussian(nlive, mean, cov, bounds=None):