lint

michaelfeil · Dec 10, 2024 · b1eae04 · b1eae04
1 parent 103cb6f
commit b1eae04
Showing 1 changed file with 0 additions and 2 deletions.
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
@@ -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