def get_data_files(data_dir):
    """ Retrieves a list from data_files to train/test with from a txt file.

    Args:
        data_dir (string): the path to the txt file
    """
    data_files = [x.strip() for x in open(data_dir).readlines()]
    return data_files