fixed numpy bug in three_level_atom module