Merge pull request #543 from roboflow/bugfix-keypoint-detection-block

Bugfix: keypoint detection model block
roboflow · Jul 20, 2024 · 6e6d1d7 · 6e6d1d7
2 parents 2bef3a2 + f207a99
commit 6e6d1d7
Show file tree

Hide file tree

Showing 3 changed files with 4 additions and 4 deletions.
diff --git a/inference/core/version.py b/inference/core/version.py
@@ -1,4 +1,4 @@
-__version__ = "0.15.0"
+__version__ = "0.15.1"
 
 
 if __name__ == "__main__":

diff --git a/inference/core/workflows/constants.py b/inference/core/workflows/constants.py
@@ -6,7 +6,7 @@
 ROOT_PARENT_ID_KEY = "root_parent_id"
 PARENT_ID_KEY = "parent_id"
 KEYPOINTS_KEY_IN_INFERENCE_RESPONSE = "keypoints"
-KEYPOINTS_CLASS_NAME_KEY_IN_INFERENCE_RESPONSE = "class"
+KEYPOINTS_CLASS_NAME_KEY_IN_INFERENCE_RESPONSE = "class_name"
 KEYPOINTS_CLASS_ID_KEY_IN_INFERENCE_RESPONSE = "class_id"
 KEYPOINTS_CONFIDENCE_KEY_IN_INFERENCE_RESPONSE = "confidence"
 KEYPOINTS_CLASS_ID_KEY_IN_SV_DETECTIONS = "keypoints_class_id"

diff --git a/tests/workflows/unit_tests/core_steps/common/test_utils.py b/tests/workflows/unit_tests/core_steps/common/test_utils.py
@@ -222,8 +222,8 @@ def test_add_inference_keypoints_to_sv_detections() -> None:
     inference_prediction = [
         {
             "keypoints": [
-                {"x": 10, "y": 20, "class": "a", "confidence": 0.3, "class_id": 1},
-                {"x": 20, "y": 30, "class": "b", "confidence": 0.4, "class_id": 0},
+                {"x": 10, "y": 20, "class_name": "a", "confidence": 0.3, "class_id": 1},
+                {"x": 20, "y": 30, "class_name": "b", "confidence": 0.4, "class_id": 0},
             ]
         },
         {