fix

Author: ydshieh

src/transformers/trainer.py

@@ -16,6 +16,7 @@
 The Trainer class, to easily train a 🤗 Transformers from scratch or finetune it on a new task.
 """
 
+import contextlib
 import functools
 import glob
 import inspect