Method save

ADHMR/lib/utils/cache.py:19–31

(cls, save_path, data_list)

    @classmethod
    def save(cls, save_path, data_list):
        assert save_path is not None, 'save_path is None'
        data_len = len(data_list)
        cache = {}
        for i, data in enumerate(data_list):
            cache[str(i)] = data
        assert len(cache) == data_len
        # update meta
        cache.update({
            'data_len': data_len,})

        np.savez_compressed(save_path, **cache)
        print(f'Cache saved to {save_path}.')

    def __len__(self):
        return self.data_len