From a57274aa480a4bbbe5d2573d0dae1444c000e363 Mon Sep 17 00:00:00 2001
From: "Wesley A. Cheng" <15952538+wesleyacheng@users.noreply.github.com>
Date: Mon, 28 Mar 2022 21:10:56 -0700
Subject: [PATCH] missing trainer import

---
 docs/source/training.mdx | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/docs/source/training.mdx b/docs/source/training.mdx
index e6c4b146fac6..0d3648a92551 100644
--- a/docs/source/training.mdx
+++ b/docs/source/training.mdx
@@ -121,7 +121,7 @@ Call `compute` on `metric` to calculate the accuracy of your predictions. Before
 If you'd like to monitor your evaluation metrics during fine-tuning, specify the `evaluation_strategy` parameter in your training arguments to report the evaluation metric at the end of each epoch:
 
 ```py
->>> from transformers import TrainingArguments
+>>> from transformers import TrainingArguments, Trainer
 
 >>> training_args = TrainingArguments(output_dir="test_trainer", evaluation_strategy="epoch")
 ```