moved gravity weighting to where it should live in nn_weights.py