diff --git a/tests/generation/test_candidate_generator.py b/tests/generation/test_candidate_generator.py
index 3a50a963a9a2..5edb909446d5 100644
--- a/tests/generation/test_candidate_generator.py
+++ b/tests/generation/test_candidate_generator.py
@@ -4,7 +4,7 @@
 from unittest.mock import MagicMock
 
 import torch
-
+import torch
 from transformers import AutoConfig, AutoModelForCausalLM, AutoTokenizer, GenerationConfig, pipeline
 from transformers.generation.candidate_generator import (
     AssistantToTargetTranslator,