add speaker information to SSC

neuromorphs · Jun 14, 2023 · c36b017 · c36b017
1 parent fb78b06
commit c36b017
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/tonic/datasets/hsd.py b/tonic/datasets/hsd.py
@@ -106,6 +106,7 @@ def __init__(
     def speaker(self):
         return self._speaker[()]
 
+
 class SSC(HSD):
     """`Spiking Speech Commands <https://zenkelab.org/resources/spiking-heidelberg-datasets-shd/>`_
     ::
@@ -166,3 +167,4 @@ def __init__(
 
         file = h5py.File(os.path.join(self.location_on_system, self.data_filename), "r")
         self.classes = file["extra/keys"][()]
+        self._speaker = file["extra/speaker"][()]