def __getitem__()

in src/dataset.py [0:0]

• 13 lines of code
• 1 McCabe index (conditional complexity)

    def __getitem__(self, idx):
        '''
        :param idx: index of the chunk to be returned
        :return: mono audio as 1 x T tensor
                 binaural audio as 2 x T tensor
                 relative rx/tx position as 7 x K tensor, where K = T / 400 (120Hz tracking vs. 48000Hz audio)
        '''
        subject = self.chunks[idx]['subject']
        offset = self.chunks[idx]['offset']
        mono = self.mono[subject][:, offset:offset+self.chunk_size]
        binaural = self.binaural[subject][:, offset:offset+self.chunk_size]
        view = self.view[subject][:, offset//400:(offset+self.chunk_size)//400]
        return mono, binaural, view