It is completely fine for your pup to come to class in a cone, as long as they are motivated and able to participate, without risk of injuring themselves further. If the cone gets in the way, taking it off during active working times should be okay as long as you are carefully watching for any action that would damage the healing sites.
Please check with your vet if you are unsure!