From 3ba32721c0a51d31a32a1267643b76d2e7584dba Mon Sep 17 00:00:00 2001 From: Dick Ameln Date: Wed, 18 May 2022 17:08:17 +0200 Subject: [PATCH] use standard visualizer instead of anomaly visualizer for anomaly tasks --- .../usecases/exportable_code/demo/demo_package/utils.py | 7 +++---- 1 file changed, 3 insertions(+), 4 deletions(-) diff --git a/ote_sdk/ote_sdk/usecases/exportable_code/demo/demo_package/utils.py b/ote_sdk/ote_sdk/usecases/exportable_code/demo/demo_package/utils.py index 17a7e395a9d..abd83c3b123 100644 --- a/ote_sdk/ote_sdk/usecases/exportable_code/demo/demo_package/utils.py +++ b/ote_sdk/ote_sdk/usecases/exportable_code/demo/demo_package/utils.py @@ -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: @@ -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")