cleanup learn_weights a tiny bit
cleanup learn_weights a tiny bit

selective-inference/selective-inference,selective-inference/selective-inference,selective-inference/selective-inference,selective-inference/selective-inference