def __getitem__()

in tbsm_data_pytorch.py [0:0]

• 8 lines of code
• 6 McCabe index (conditional complexity)

    def __getitem__(self, index):

        if isinstance(index, slice):
            return [
                self[idx] for idx in range(
                    index.start or 0, index.stop or len(self), index.step or 1
                )
            ]

        return self.X_cat[:, index, :], self.X_int[:, index, :], self.y[index]