Anyone brought their LO to cemetery before? To pray for the deceased? Is it okay?

I would never bring them as I was told by my friend that after she allowed her baby to attend her grandmother's funeral, her baby got high fever the next day. But it's really a dilemma because it's between 2 loved ones. Maybe if the deceased have already passed away long ago, let the LO visit after he or she turns older? I'm sure the deceased will understand.