From b1eae04a41bde325b03103cadade80edc65c9930 Mon Sep 17 00:00:00 2001 From: michaelfeil <63565275+michaelfeil@users.noreply.github.com> Date: Tue, 10 Dec 2024 00:21:16 -0800 Subject: [PATCH] lint --- .../unit_test/transformer/classifier/test_optimum_classifier.py | 2 -- 1 file changed, 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 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