Skip to content

declearn.messaging.EvaluationReply

Bases: Message

Client-emitted results from a local evaluation round.

Source code in declearn/messaging/_base.py
 83
 84
 85
 86
 87
 88
 89
 90
 91
 92
 93
 94
 95
 96
 97
 98
 99
100
@dataclasses.dataclass
class EvaluationReply(Message):
    """Client-emitted results from a local evaluation round."""

    typekey = "eval_reply"

    loss: float
    n_steps: int
    t_spent: float
    metrics: Dict[str, MetricState] = dataclasses.field(default_factory=dict)

    def to_kwargs(
        self,
    ) -> Dict[str, Any]:
        # Undo recursive dict-conversion of dataclasses.
        kwargs = super().to_kwargs()
        kwargs["metrics"] = self.metrics
        return kwargs