import pathlib
from typing import Tuple


def prepare_train_valid(
        input_tfr: pathlib.Path) -> Tuple[pathlib.Path, pathlib.Path]:
    """Get all filennames for train and validation files.

    Args:
        input_tfr (pathlib.Path): Name of input directory.

    Returns:
        Tuple[List[pathlib.Path], List[pathlib.Path]]: Add train and validation
            files to seperate lists.
    """
    train_files = input_tfr.joinpath('training.tfrecords_v2')
    valid_files = input_tfr.joinpath('validate.tfrecords_v2')
    return train_files, valid_files