diff --git a/libs/infinity_emb/tests/unit_test/transformer/classifier/test_optimum_classifier.py b/libs/infinity_emb/tests/unit_test/transformer/classifier/test_optimum_classifier.py
index 9c4a3795..95b4104d 100644
--- a/libs/infinity_emb/tests/unit_test/transformer/classifier/test_optimum_classifier.py
+++ b/libs/infinity_emb/tests/unit_test/transformer/classifier/test_optimum_classifier.py
@@ -1,6 +1,4 @@
-import torch
 from transformers.pipelines import pipeline  # type: ignore
-from optimum.onnxruntime import ORTModelForSequenceClassification
 from infinity_emb.args import EngineArgs
 
 from infinity_emb.transformer.classifier.optimum import OptimumClassifier