"""
parameters only to run the evaluate_predictions.py script
"""

from dataclasses import dataclass
from conf._util import return_factory
from typing import Union


@dataclass
class EvaluationParams:
    folder_predictions: list[str] = return_factory(["/path/to/images"])
    loadfunction: str = ""
    """
    controlnet_brats2020_1234_0
    """

    diversity_nb: int = 10
    device: str = "cuda"
