def generate_model_card()

in jat/utils.py [0:0]

• 19 lines of code
• 5 McCabe index (conditional complexity)

def generate_model_card(model_name: str, evaluations: Optional[Dict[str, List[float]]] = None) -> ModelCard:
    """
    Generate a ModelCard from a template.

    Args:
        model_name (`str`):
            Model name.
        evaluations (`Dict[str, List[float]]`):
            Dictionary containing the evaluation results for each task.

    Returns:
        `ModelCard`:
            A ModelCard object.
    """
    tags = ["reinforcement-learning"]
    if evaluations is not None:
        tags.extend(evaluations.keys())
    card_data = ModelCardData(
        tags=tags,
        eval_results=generate_rl_eval_results(evaluations) if evaluations is not None else None,
        model_name=model_name,
        datasets="jat-project/jat-dataset",
        pipeline_tag="reinforcement-learning",
    )
    card = ModelCard.from_template(
        card_data,
        template_path="templates/model_card.md",
        model_name=model_name,
        model_id="Jat",
        tasks=[PRETTY_TASK_NAMES[task_name] for task_name in evaluations.keys()] if evaluations is not None else [],
    )
    return card