12b309342f718da259eabd189b74269932f485fa,nilmtk/stats/goodsectionsresults.py,GoodSectionsResults,import_from_cache,#GoodSectionsResults#,109

Before Change

    def import_from_cache(self, cached_stat, sections):
        usable_sections_from_cache = pd.DataFrame()
        for section in sections:
            try:
                row = cached_stat.loc[section.start]
            except KeyError:
                pass
            else:
                end_time = row["end"]
                if end_time == section.end:
                    usable_sections_from_cache = (
                        usable_sections_from_cache.append(row))

        grouped_by_index = usable_sections_from_cache.groupby(level=0)
        for name, group in grouped_by_index:
            assert group["end"].unique().size == 1
            timeframes = [TimeFrame(row["section_start"], row["section_end"])

After Change

        for name, group in grouped_by_index:
            assert group["end"].unique().size == 1
            timeframe = TimeFrame(name, group["end"].iloc[0])
            if timeframe in sections:
                timeframes = [TimeFrame(row["section_start"], row["section_end"])
                              for _, row in group.iterrows()]
                self.append(timeframe, {"sections": [timeframes]})

Instances