def __getitem__()

in optimum/amd/brevitas/data_utils.py [0:0]

• 5 lines of code
• 3 McCabe index (conditional complexity)

    def __getitem__(self, idx):
        if self.device is not None:
            return {name: recursive_to_device(val, self.device) for name, val in self.data[idx].items()}
        else:
            return self.data[idx]