From da4404e8750d9a69efd4fb0b33f06f4e341b5a55 Mon Sep 17 00:00:00 2001 From: Raif Olson Date: Thu, 28 Mar 2024 08:38:50 -0400 Subject: [PATCH] More pythonic version of change --- supervision/detection/utils.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/supervision/detection/utils.py b/supervision/detection/utils.py index 762f06c7..25a2a37c 100644 --- a/supervision/detection/utils.py +++ b/supervision/detection/utils.py @@ -849,7 +849,7 @@ def get_data_item( subset_data[key] = [value[i] for i in index] elif isinstance(index, np.ndarray): if index.dtype == bool: - subset_data[key] = [value[i] for i in range(len(index)) if index[i]] + subset_data[key] = [value[i] for i, index_value in enumerate(index) if index_value] else: subset_data[key] = [value[i] for i in index] elif isinstance(index, int):