bc1ca6bc6d1c28c780e348321e8e95d5b3f91794,nussl/deep/datasets/base_dataset.py,BaseDataset,write_to_cache,#BaseDataset#,192

Before Change

        with open(os.path.join(self.cache, filename), "wb") as f:
            // for key in data:
            //     data_dict[key] = data_dict[key].astype(PRECISION[self.save_precision].value)
            pickle.dump(data_dict, f)

    def load_from_cache(self, filename: str):
        with open(os.path.join(self.cache, filename), "rb") as f:
            data = pickle.load(f)

After Change

        self.cache_populated = True

    def write_to_cache(self, data_dict, i):
        self.cache_dataset[i] = data_dict

    def load_from_cache(self, i):
        data = self.cache_dataset[i]

Instances