diff --git a/summit/multiview_platform/utils/dataset.py b/summit/multiview_platform/utils/dataset.py index 600a06618169570e18e400901742f303c53f3402..eb00dfd580ce3cecf3d91702c385297b52190303 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"])