Baby has gotten some of the basics of dancing down, but there is still a long way to go to be prepared for the hotel audition she has to dance with Johnny. But something is happening during the intense training sessions: their bond is growing closer and getting more intimate with their close contact.
This is almost a Dating Montage in some respects. For one, notice that as the montage goes on, they keep wearing less clothes. But, since the montage still focuses on the actual dance preparation for the contest under the direction of a mentor, it is still a Training Montage.