Skip to content
Merged
Empty file.
6 changes: 5 additions & 1 deletion abacusnbody/hod/menv.py
Original file line number Diff line number Diff line change
Expand Up @@ -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,
Expand Down Expand Up @@ -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
Expand Down
Loading