From 94131ff56b252ddf29934eda3790abee8fe5c5c0 Mon Sep 17 00:00:00 2001 From: Soumik Rakshit <19soumik.rakshit96@gmail.com> Date: Tue, 22 Apr 2025 18:13:34 +0000 Subject: [PATCH] add: support for non-str labels --- supervision/annotators/core.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/supervision/annotators/core.py b/supervision/annotators/core.py index 1b5b9bee..d42e46d3 100644 --- a/supervision/annotators/core.py +++ b/supervision/annotators/core.py @@ -1130,7 +1130,7 @@ class LabelAnnotator(BaseAnnotator): detections: Detections, custom_labels: Optional[List[str]] ) -> List[str]: if custom_labels is not None: - return custom_labels + return [str(label) for label in custom_labels] labels = [] for idx in range(len(detections)):