fix(aliases): resolve public yolo26-sem model aliases for v0 clients (#2394)

Author: leeclemnet

inference/models/aliases.py

@@ -88,6 +88,19 @@
     "yolo26x-pose-640": "coco-pose-detection/16",
 }
 
+YOLO26_SEM_ALIASES = {
+    "yolov26n-sem-1024": "yolo26-pretrains/yolo26n-sem",
+    "yolov26s-sem-1024": "yolo26-pretrains/yolo26s-sem",
+    "yolov26m-sem-1024": "yolo26-pretrains/yolo26m-sem",
+    "yolov26l-sem-1024": "yolo26-pretrains/yolo26l-sem",
+    "yolov26x-sem-1024": "yolo26-pretrains/yolo26x-sem",
+    "yolo26n-sem-1024": "yolo26-pretrains/yolo26n-sem",
+    "yolo26s-sem-1024": "yolo26-pretrains/yolo26s-sem",
+    "yolo26m-sem-1024": "yolo26-pretrains/yolo26m-sem",
+    "yolo26l-sem-1024": "yolo26-pretrains/yolo26l-sem",
+    "yolo26x-sem-1024": "yolo26-pretrains/yolo26x-sem",
+}
+
 YOLOV11_ALIASES = {
     **YOLOV11_ALIASES,
     **{k.replace("yolov11", "yolo11"): v for k, v in YOLOV11_ALIASES.items()},
@@ -172,6 +185,7 @@
     **YOLO26_ALIASES,
     **YOLO26_SEG_ALIASES,
     **YOLO26_POSE_ALIASES,
+    **YOLO26_SEM_ALIASES,
 }
 
 

inference_sdk/http/utils/aliases.py

@@ -60,6 +60,19 @@
     "yolo26x-pose-640": "coco-pose-detection/16",
 }
 
+YOLO26_SEM_ALIASES = {
+    "yolov26n-sem-1024": "yolo26-pretrains/yolo26n-sem",
+    "yolov26s-sem-1024": "yolo26-pretrains/yolo26s-sem",
+    "yolov26m-sem-1024": "yolo26-pretrains/yolo26m-sem",
+    "yolov26l-sem-1024": "yolo26-pretrains/yolo26l-sem",
+    "yolov26x-sem-1024": "yolo26-pretrains/yolo26x-sem",
+    "yolo26n-sem-1024": "yolo26-pretrains/yolo26n-sem",
+    "yolo26s-sem-1024": "yolo26-pretrains/yolo26s-sem",
+    "yolo26m-sem-1024": "yolo26-pretrains/yolo26m-sem",
+    "yolo26l-sem-1024": "yolo26-pretrains/yolo26l-sem",
+    "yolo26x-sem-1024": "yolo26-pretrains/yolo26x-sem",
+}
+
 RFDETR_ALIASES = {
     "rfdetr-base": "coco/36",
     # "rfdetr-large": "coco/37", deprecated
@@ -155,6 +168,7 @@
     **YOLO26_ALIASES,
     **YOLO26_SEG_ALIASES,
     **YOLO26_POSE_ALIASES,
+    **YOLO26_SEM_ALIASES,
 }
 
 OCR_ENDPOINTS = {

tests/inference/unit_tests/models/test_aliases.py

@@ -15,3 +15,12 @@ def test_resolve_roboflow_model_alias_when_registry_hit_should_not_happen() -> N
 
     # then
     assert result == "my_project/3"
+
+
+def test_resolve_roboflow_model_alias_for_yolo26_sem_public_models() -> None:
+    for size in ["n", "s", "m", "l", "x"]:
+        # when
+        result = resolve_roboflow_model_alias(model_id=f"yolo26{size}-sem-1024")
+
+        # then
+        assert result == f"yolo26-pretrains/yolo26{size}-sem"

tests/inference_sdk/unit_tests/http/utils/test_aliases.py

@@ -15,3 +15,12 @@ def test_resolve_roboflow_model_alias_when_registry_hit_should_not_happen() -> N
 
     # then
     assert result == "my_project/3"
+
+
+def test_resolve_roboflow_model_alias_for_yolo26_sem_public_models() -> None:
+    for size in ["n", "s", "m", "l", "x"]:
+        # when
+        result = resolve_roboflow_model_alias(model_id=f"yolo26{size}-sem-1024")
+
+        # then
+        assert result == f"yolo26-pretrains/yolo26{size}-sem"