diff --git a/ote_sdk/ote_sdk/entities/model_template.py b/ote_sdk/ote_sdk/entities/model_template.py index 8adfa2f2786..b03f9dbc6a0 100644 --- a/ote_sdk/ote_sdk/entities/model_template.py +++ b/ote_sdk/ote_sdk/entities/model_template.py @@ -477,6 +477,7 @@ def __init__(self) -> None: TaskType.ANOMALY_DETECTION, TaskType.ANOMALY_CLASSIFICATION, TaskType.ANOMALY_SEGMENTATION, + TaskType.ROTATED_DETECTION, ) diff --git a/ote_sdk/ote_sdk/tests/entities/test_model_template.py b/ote_sdk/ote_sdk/tests/entities/test_model_template.py index a9a2f12f1f9..fa8e6590e42 100644 --- a/ote_sdk/ote_sdk/tests/entities/test_model_template.py +++ b/ote_sdk/ote_sdk/tests/entities/test_model_template.py @@ -1049,6 +1049,7 @@ def test_task_type_constants(self): TaskType.ANOMALY_DETECTION, TaskType.ANOMALY_CLASSIFICATION, TaskType.ANOMALY_SEGMENTATION, + TaskType.ROTATED_DETECTION, )