It isn't a punishment if the AI is doing the work. The goal is to make students utilize the skills they will need now that AI can supplement their abilities.
But are AIs now _required_? If you're tailoring the class to allow growth potential for students using AI, what happens to the students who cannot use it (for whatever reason)?
It's a bit similar to making a class harder because some students are getting extra help via private tutoring.