HepNet save fails because of missing keras dependence