From 5c212bfc34dd19b3b5efdd5e76c6fa8f6ea52a69 Mon Sep 17 00:00:00 2001
From: Baptiste Bauvin <baptiste.bauvin@lis-lab.fr>
Date: Tue, 28 Feb 2023 11:30:47 -0500
Subject: [PATCH] Support numpy bytes
---
summit/multiview_platform/utils/dataset.py | 2 +-
1 file changed, 1 insertion(+), 1 deletion(-)
diff --git a/summit/multiview_platform/utils/dataset.py b/summit/multiview_platform/utils/dataset.py
index 600a0661..eb00dfd5 100644
--- a/summit/multiview_platform/utils/dataset.py
+++ b/summit/multiview_platform/utils/dataset.py
@@ -503,7 +503,7 @@ class HDF5Dataset(Dataset):
seleted labels' names
"""
selected_labels = self.get_labels(sample_indices)
- if type(self.dataset["Labels"].attrs["names"][0]) == bytes:
+ if type(self.dataset["Labels"].attrs["names"][0]) in [bytes, np.bytes_]:
return [label_name.decode("utf-8")
for label, label_name in
enumerate(self.dataset["Labels"].attrs["names"])
--
GitLab