yohai/phigrape
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity

run directory now separate from source directory

Browse source
This commit is contained in:
Yohai Meiron 2021-03-17 20:37:12 -04:00
parent 11c0db23a3
commit 0a2344563d
3 changed files with 11 additions and 7 deletions
Download patch file Download diff file Expand all files Collapse all files

2
init.py
View file

@ -91,6 +91,8 @@ def gen_mask(particle_list, frac):
mask = np.ones(N, dtype=int)
elif frac==1:
mask = np.zeros(N, dtype=int)
elif (frac < 0) or (1 < frac):
raise RuntimeError('Fraction has to be between 0 and 1')
else:
X = particle_list[:,:3]
V = particle_list[:,3:]