[trainer] apex test fix
there was a small mistake in a test of https://github.com/huggingface/transformers/pull/18961, this PR fixes it.

the main CI doesn't have apex installed that's why it missed it.

Thank you, @ydshieh for the heads up about the breakage

_The documentation is not available anymore as the PR was closed or merged._