utils.py 文件源码

def __init__(self, data_dir):
        self.data_dir = data_dir

        metadata_filename = os.path.join(self.data_dir, "metadata.joblib")
        if os.path.exists(metadata_filename):
            # self.tasks, self.metadata_df = joblib.load(metadata_filename)
            self.tasks, self.metadata_df = load_from_disk(metadata_filename)
        else:
            raise ValueError("No metadata found on disk")