I used to shoot quite a lot of pics some years back of kids on a PADI Bubblemaker course. I used a D300 and the Tokina 10-17.
Although I'd agree with Undertow on using a non-fisheye, I did find that using the Tokina did produce a good number of very useable and fun images underwater. Sure, some had weird perspectives but, perhaps inevitably, when photographing people, lots of the images were rubbish - and some were terrific. I also did a couple of model-type photoshoots in a pool and, again, used the 10-17 with good results.
Yep, the 10-17 above the water, not so good.
I think you are right though, relax and enjoy the party - stick your camera on multi-shot release and shoot away. The kids - and their parents - love the underwater shots.
One final, rather depressing, thought though: I was photographing (and ran) the Bubblemaker course back in the 00s. Given all the issues now raised about working with kids, I'd be much more wary and make absolutely sure that parents were ok with it. Even taking pics of my grandkids (yeah, I'm that old) now in the park, I feel, sadly, a bit uncomfortable when there are other kids and parents around. All a bit sad really.