Added the necessay import of module (#30804)

huggingface · May 24, 2024 · 9bb3087 · 9bb3087
1 parent 95d9858
commit 9bb3087
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/docs/source/en/training.md b/docs/source/en/training.md
@@ -186,6 +186,7 @@ so we can just convert that directly to a NumPy array without tokenization!
 
 ```py
 from transformers import AutoTokenizer
+import numpy as np
 
 tokenizer = AutoTokenizer.from_pretrained("google-bert/bert-base-cased")
 tokenized_data = tokenizer(dataset["sentence"], return_tensors="np", padding=True)