import numpy as np
import os

def load_dataset(config):
    dataset_name = config['basic']['dataset']
    node_num = config['data']['num_nodes']
    input_dim = config['data']['input_dim']
    data = None
    match dataset_name:
        case 'EcoSolar':
            data_path = os.path.join('./data/EcoSolar.npy')
            data = np.load(data_path)[:, :node_num, :input_dim]
    return data