prepare.py

import pickle
import datasets
import os
import pandas as pd

from datasets import load_dataset_builder

if __name__ == "__main__":
    
    builder = load_dataset_builder('renumics/emodb-enriched')

    
    if os.path.exists(builder._cache_dir):
        # Load dataset from cache
        ds = datasets.load_dataset('renumics/emodb-enriched', split='all')
        print("Dataset loaded from cache.")
    else:
        # Load dataset using datasets.load_dataset()
        ds = datasets.load_dataset('renumics/emodb-enriched', split='all')
        print("Dataset loaded using datasets.load_dataset().")