def returnIndicesTest(splits):
    """Takes the splits and return the train, test and dev splits."""
    test = []
    train = []
    dev = []

    for split in splits.keys():
        if split in [0, 1]:
            test += splits[split]
        if split in [2]:
            dev += splits[split]
        else:
            train += splits[split]

    return train, test, dev