From 1bb0d37767406d12881490e2003cf50f6bf1238a Mon Sep 17 00:00:00 2001
From: kkeroo <61207502+kkeroo@users.noreply.github.com>
Date: Fri, 30 Aug 2024 16:32:08 +0200
Subject: [PATCH] Fix tests.

---
 tests/unittests/test_creators/test_detections.py | 9 ++++-----
 1 file changed, 4 insertions(+), 5 deletions(-)

diff --git a/tests/unittests/test_creators/test_detections.py b/tests/unittests/test_creators/test_detections.py
index 8747a0f..761b49f 100644
--- a/tests/unittests/test_creators/test_detections.py
+++ b/tests/unittests/test_creators/test_detections.py
@@ -5,8 +5,8 @@
 import pytest
 
 from depthai_nodes.ml.messages import (
-    ImgDetectionsWithKeypoints,
-    ImgDetectionWithKeypoints,
+    ImgDetectionExtended,
+    ImgDetectionsExtended,
 )
 from depthai_nodes.ml.messages.creators.detection import create_detection_message
 
@@ -172,10 +172,9 @@ def test_bboxes_scores_keypoints():
 
     message = create_detection_message(bboxes, scores, None, keypoints)
 
-    assert isinstance(message, ImgDetectionsWithKeypoints)
+    assert isinstance(message, ImgDetectionsExtended)
     assert all(
-        isinstance(detection, ImgDetectionWithKeypoints)
-        for detection in message.detections
+        isinstance(detection, ImgDetectionExtended) for detection in message.detections
     )
 
     for i, detection in enumerate(message.detections):