Skip to content

Commit

Permalink
Merge pull request #1113 from openvinotoolkit/da/anomaly-exportable-c…
Browse files Browse the repository at this point in the history
…ode-fix

[ANOMALY] Fix: exportable code for anomaly tasks
  • Loading branch information
goodsong81 authored May 31, 2022
2 parents 56e6624 + 3ba3272 commit 02d724a
Showing 1 changed file with 3 additions and 4 deletions.
Original file line number Diff line number Diff line change
Expand Up @@ -14,7 +14,7 @@
from ote_sdk.usecases.exportable_code.prediction_to_annotation_converter import (
create_converter,
)
from ote_sdk.usecases.exportable_code.visualizers import AnomalyVisualizer, Visualizer
from ote_sdk.usecases.exportable_code.visualizers import Visualizer


def get_model_path(path: Optional[Path]) -> Path:
Expand Down Expand Up @@ -55,12 +55,11 @@ def create_output_converter(task_type: TaskType, labels: LabelSchemaEntity):
return create_converter(converter_type, labels)


def create_visualizer(task_type: TaskType):
def create_visualizer(_task_type: TaskType):
"""
Create visualizer according to kind of task
"""

if task_type.is_anomaly:
return AnomalyVisualizer(window_name="Result")
# TODO: use anomaly-specific visualizer for anomaly tasks

return Visualizer(window_name="Result")

0 comments on commit 02d724a

Please sign in to comment.