def get_dataset(training_info_txt: str) -> str:
    """get dataset name from training_info.txt

    Args:
        training_info_txt (str): path to training_info.txt in checkpoints dir

    Returns:
        str: dataset name
    """
    with open(training_info_txt, "r") as f:
        dataset = f.read().split("\n")[0].split(":")[-1].replace(" ", "")
        return dataset