use standard visualizer instead of anomaly visualizer for anomaly tasks

openvinotoolkit · May 18, 2022 · 3ba3272 · 3ba3272
1 parent c49a6fd
commit 3ba3272
Showing 1 changed file with 3 additions and 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
@@ -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")