def METHOD_NAME(
    self, parameters: NDArrays, config: Dict[str, Scalar]
) -> Tuple[NDArrays, int, Dict]:
    """Implements distributed fit function for a given client."""
    self.set_parameters(parameters)
    model.train(
        self.net,
        self.trainloader,
        self.device,
        epochs=self.num_epochs,
        learning_rate=self.learning_rate,
    )
    return self.get_parameters({}), len(self.trainloader), {}