def load(self, path, fmt=None): if not fmt: if filename_is_json(path): self.load_json(path) else: self.load_pickle(path) elif fmt in ['json', 'jsonl']: self.load_json(path) else: self.load_pickle(path)
def save(self, path, fmt=None): if not fmt: if filename_is_json(path): self.save_json(path) else: self.save_pickle(path) elif fmt in ['json', 'jsonl']: self.save_json(path) else: self.save_pickle(path)