declearn.messaging.TrainReply

Bases: Message

Client-emitted results from a local training round.

Source code in declearn/messaging/_base.py

@dataclasses.dataclass
class TrainReply(Message):
    """Client-emitted results from a local training round."""

    typekey = "train_reply"

    n_epoch: int
    n_steps: int
    t_spent: float
    updates: ModelUpdates
    aux_var: Dict[str, AuxVar]

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