You only need a 3rd class to act as PIC, training flight or not.

The only time 2nd class is needed is when the flight is for
compensation or hire, but excludes any flight excercising privileges of
a CFI certificate. At least thats my understanding.