from pathlib import Path


def setup_results_dir(experiment_name):
    """Create the directories for the results."""
    root = Path(__file__).parent.parent
    result_path = root / "batchpredict"
    results_dir = (
        result_path
        / experiment_name
    )
    results_dir.mkdir(parents=True, exist_ok=True)
    print(f"Writing results to {results_dir}")
    return results_dir