From 8e32e56d04911d62cd97578fac6b46ada5383178 Mon Sep 17 00:00:00 2001 From: kkeroo <61207502+kkeroo@users.noreply.github.com> Date: Tue, 10 Dec 2024 17:40:48 +0100 Subject: [PATCH] Fix validation in cluster creation. --- depthai_nodes/ml/messages/creators/clusters.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/depthai_nodes/ml/messages/creators/clusters.py b/depthai_nodes/ml/messages/creators/clusters.py index 4a1a382d..8f2e4696 100644 --- a/depthai_nodes/ml/messages/creators/clusters.py +++ b/depthai_nodes/ml/messages/creators/clusters.py @@ -25,9 +25,9 @@ def create_cluster_message(clusters: List[List[List[Union[float, int]]]]) -> Clu if not isinstance(cluster, list): raise TypeError(f"All clusters must be of type List, got {type(cluster)}") for point in cluster: - if not isinstance(point, list): + if not isinstance(point, tuple) and not isinstance(point, list): raise TypeError( - f"All points in clusters must be of type List, got {type(point)}" + f"All points in clusters must be of type tuple or list, got {type(point)}" ) if len(point) != 2: raise ValueError(f"Each point must have 2 values, got {len(point)}")