from gen_utils import read_json

def load_motionfix(path_to_data):
    dataset = read_json(path_to_data + '/amt_motionfix_latest_final.json')
    splits = read_json(path_to_data + '/splits_final.json')

    train_data = {}
    val_data = {}
    test_data = {}

    # Fill each dictionary with the corresponding data
    for key in splits['train']:
        if key in dataset:
            train_data[key] = dataset[key]

    for key in splits['val']:
        if key in dataset:

            val_data[key] = dataset[key]

    for key in splits['test']:
        if key in dataset:
            test_data[key] = dataset[key]
    validation_test_data = {**val_data, **test_data}

    return train_data, validation_test_data