expand ModelTemplate.is_global

openvinotoolkit · Mar 17, 2022 · 1c1baa6 · 1c1baa6
1 parent 745ec57
commit 1c1baa6
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/ote_sdk/ote_sdk/entities/model_template.py b/ote_sdk/ote_sdk/entities/model_template.py
@@ -422,7 +422,10 @@ def is_task_global(self) -> bool:
         """
         Returns ``True`` if the task is global task i.e. if task produces global labels
         """
-        return self.task_type in [TaskType.CLASSIFICATION]
+        return self.task_type in (
+            TaskType.CLASSIFICATION,
+            TaskType.ANOMALY_CLASSIFICATION,
+        )
 
 
 class NullModelTemplate(ModelTemplate):