Source code for fl4health.parameter_exchange.parameter_exchanger_base

from abc import ABC, abstractmethod
from typing import TypeVar

import torch.nn as nn
from flwr.common.typing import Config, NDArrays



[docs]
class ParameterExchanger(ABC):

[docs]
    @abstractmethod
    def push_parameters(
        self, model: nn.Module, initial_model: nn.Module | None = None, config: Config | None = None
    ) -> NDArrays:
        raise NotImplementedError



[docs]
    @abstractmethod
    def pull_parameters(self, parameters: NDArrays, model: nn.Module, config: Config | None = None) -> None:
        raise NotImplementedError




ExchangerType = TypeVar("ExchangerType", bound=ParameterExchanger)