Merge pull request #1062 from rolson24/fix-Detections-call-bug

fix Detections.get_data_item() bug (#1061)
This commit is contained in:
Piotr Skalski authored and GitHub committed 2024-03-28 13:40:57 +01:00
commit 69563b8bbb
1 file changed
+8 -1
+8 -1
View File
@@ -845,8 +845,15 @@ def get_data_item(
elif isinstance(value, list):
if isinstance(index, slice):
subset_data[key] = value[index]
elif isinstance(index, (list, np.ndarray)):
elif isinstance(index, list):
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, index_value in enumerate(index) if index_value
]
else:
subset_data[key] = [value[i] for i in index]
elif isinstance(index, int):
subset_data[key] = [value[index]]
else: