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):