fa926b8278a3b635d929e0fe6c63ac2a8983a406,lib/prody/dynamics/nma.py,NMA,__getitem__,#NMA#Any#,58

Before Change

        if self._n_modes == 0:
            raise ValueError("{0} modes are not calculated, try "
                             "calcModes() method".format(str(self)))
        indices = self._indices
        if indices is None:
            self._indices = indices = np.arange(self._n_modes)
        try:
            indices = indices[index]
        except IndexError as err:
            raise IndexError(str(err))
        try:
            length = len(indices)
        except TypeError:            
            return Mode(self, indices)

After Change

        if isinstance(index, int):
            return self._getMode(index)
        elif isinstance(index, slice):
            indices = np.arange(*index.indices(len(self)))
            if len(indices) > 0:
                return ModeSet(self, indices)
        elif isinstance(index, (list, tuple)):
            for i in index:

Instances