dcbfeceae637c576a1494e230375dbd682954e3d,batchflow/components.py,ComponentsDict,crop,#ComponentsDict#Any#,146

Before Change

         Crops from data in accordance with indices 
        if self.data is not None and self.indices is not None:
            new_data = {}
            components = self.components or list(self.data.keys())
            for comp in components:
                data = self.data.get(comp, None)
                data = self._get_from(data, copy)
                new_data[comp] = data

After Change

    def crop(self, indices=None):
         Crops from data in accordance with indices 
        if isinstance(self.data, pd.DataFrame):
            self.data = self.data.loc[indices]
        else:
            new_data = {}
            for comp in self.components:
                if isinstance(self.data, BaseComponents):
                    comp_data = self.data.get(comp, indices)
                else:
                    comp_data = self.data.get(comp)[indices]
                new_data[comp] = comp_data
            self.data = new_data

    @property
    def indices(self):
        return None if self._crop else self._indices

Instances