Skip to content

Commit 3e3b1f7

Browse files
ignore unused import for tests
Signed-off-by: Ashwin Vaidya <[email protected]>
1 parent b178029 commit 3e3b1f7

File tree

2 files changed

+25
-0
lines changed

2 files changed

+25
-0
lines changed

tests/cpp/precommit/prepare_data.py

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -22,6 +22,10 @@ def prepare_model(
2222
data_dir="./data",
2323
public_scope=Path(__file__).resolve().parent / "public_scope.json",
2424
):
25+
# TODO refactor this test so that it does not use eval
26+
# flake8: noqa: F401
27+
from model_api.models import ClassificationModel, DetectionModel, SegmentationModel
28+
2529
with open(public_scope, "r") as f:
2630
public_scope = json.load(f)
2731

tests/python/accuracy/test_accuracy.py

Lines changed: 21 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -6,23 +6,44 @@
66
import numpy as np
77
import onnx
88
import pytest
9+
910
from model_api.adapters.onnx_adapter import ONNXRuntimeAdapter
1011
from model_api.adapters.openvino_adapter import OpenvinoAdapter, create_core
1112
from model_api.adapters.utils import load_parameters_from_onnx
13+
14+
# TODO refactor this test so that it does not use eval
15+
# flake8: noqa: F401
1216
from model_api.models import (
1317
ActionClassificationModel,
18+
AnomalyDetection,
1419
AnomalyResult,
20+
ClassificationModel,
1521
ClassificationResult,
1622
DetectedKeypoints,
23+
DetectionModel,
1724
DetectionResult,
25+
ImageModel,
1826
ImageResultWithSoftPrediction,
1927
InstanceSegmentationResult,
28+
KeypointDetectionModel,
29+
MaskRCNNModel,
30+
PredictedMask,
2031
Prompt,
32+
SAMDecoder,
33+
SAMImageEncoder,
34+
SAMLearnableVisualPrompter,
35+
SAMVisualPrompter,
36+
SegmentationModel,
2137
VisualPromptingResult,
2238
ZSLVisualPromptingResult,
2339
add_rotated_rects,
2440
get_contours,
2541
)
42+
from model_api.tilers import (
43+
DetectionTiler,
44+
InstanceSegmentationTiler,
45+
SemanticSegmentationTiler,
46+
)
2647

2748

2849
def read_config(path: Path):

0 commit comments

Comments
 (0)