Merge pull request #212 from SBouchard01/bgs_prep

Adding BGS filter to prepare_sim

Author: lgarrison

abacusnbody/analysis/__init__.py

@@ -9,12 +9,16 @@
 import numpy as np
 from scipy.spatial import KDTree
 
+import logging
+
 from ..util import cumsum
 
 __all__ = ['do_Menv_from_tree']
 
 DEFAULT_BATCH_SIZE = 10**5
 
+logger = logging.getLogger(__name__)
+
 
 def do_Menv_from_tree(
     pos,
@@ -53,7 +57,7 @@ def do_Menv_from_tree(
     if r_outer.ndim > 0:
         r_outer = r_outer[mmask]
 
-    print('Building and querying trees for mass env calculation')
+    logger.info('Building and querying trees for mass env calculation')
     tree = KDTree(pos, boxsize=treebox)
 
     # we're taking potentially large differences, use float64