82844481da6e1f3d391541d47af0c1466679f0f0,nilmtk/metergroup.py,MeterGroup,__getitem__,#MeterGroup#,107

Before Change

            for meter in self.meters: // TODO: write unit tests for this
                if isinstance(meter, MeterGroup):
                    metergroup = meter
                    meter_ids = set([m.identifier for m in metergroup.meters
                                     if isinstance(m, ElecMeter)])
                    if meter_ids == set(key):
                        return meter
            raise KeyError(key)
        elif isinstance(key, tuple):

After Change

                        group.building() == key.building and
                        group.dataset() == key.dataset):
                        return group
            elif key.instance == 0:
                metergroup_of_building = nilmtk.global_meter_group.select(
                    building=key.building, dataset=key.dataset)
                return metergroup_of_building.mains()
            else:
                for meter in self.meters:
                    if meter.identifier == key:
                        return meter

Instances