ci(*:skip) Update client_fn args in e2e tests

e2e/bare-client-auth/client.py

@@ -1,13 +1,14 @@
 import numpy as np
 
-import flwr as fl
+from flwr.client import ClientApp, NumPyClient
+from flwr.common import Context
 
 model_params = np.array([1])
 objective = 5
 
 
 # Define Flower client
-class FlowerClient(fl.client.NumPyClient):
+class FlowerClient(NumPyClient):
     def get_parameters(self, config):
         return model_params
 
@@ -23,10 +24,10 @@ def evaluate(self, parameters, config):
         return loss, 1, {"accuracy": accuracy}
 
 
-def client_fn(cid):
+def client_fn(context: Context):
     return FlowerClient().to_client()
 
 
-app = fl.client.ClientApp(
+app = ClientApp(
     client_fn=client_fn,
 )

e2e/bare-https/client.py

@@ -2,14 +2,15 @@
 
 import numpy as np
 
-import flwr as fl
+from flwr.client import ClientApp, NumPyClient, start_client
+from flwr.common import Context
 
 model_params = np.array([1])
 objective = 5
 
 
 # Define Flower client
-class FlowerClient(fl.client.NumPyClient):
+class FlowerClient(NumPyClient):
     def get_parameters(self, config):
         return model_params
 
@@ -25,17 +26,17 @@ def evaluate(self, parameters, config):
         return loss, 1, {"accuracy": accuracy}
 
 
-def client_fn(cid):
+def client_fn(context: Context):
     return FlowerClient().to_client()
 
 
-app = fl.client.ClientApp(
+app = ClientApp(
     client_fn=client_fn,
 )
 
 if __name__ == "__main__":
     # Start Flower client
-    fl.client.start_client(
+    start_client(
         server_address="127.0.0.1:8080",
         client=FlowerClient().to_client(),
         root_certificates=Path("certificates/ca.crt").read_bytes(),

e2e/bare/client.py

@@ -2,8 +2,8 @@
 
 import numpy as np
 
-import flwr as fl
-from flwr.common import ConfigsRecord
+from flwr.client import ClientApp, NumPyClient, start_client
+from flwr.common import ConfigsRecord, Context
 
 SUBSET_SIZE = 1000
 STATE_VAR = "timestamp"
@@ -14,7 +14,7 @@
 
 
 # Define Flower client
-class FlowerClient(fl.client.NumPyClient):
+class FlowerClient(NumPyClient):
     def get_parameters(self, config):
         return model_params
 
@@ -51,16 +51,14 @@ def evaluate(self, parameters, config):
         )
 
 
-def client_fn(cid):
+def client_fn(context: Context):
     return FlowerClient().to_client()
 
 
-app = fl.client.ClientApp(
+app = ClientApp(
     client_fn=client_fn,
 )
 
 if __name__ == "__main__":
     # Start Flower client
-    fl.client.start_client(
-        server_address="127.0.0.1:8080", client=FlowerClient().to_client()
-    )
+    start_client(server_address="127.0.0.1:8080", client=FlowerClient().to_client())

e2e/docker/client.py

@@ -9,6 +9,7 @@
 from torchvision.transforms import Compose, Normalize, ToTensor
 
 from flwr.client import ClientApp, NumPyClient
+from flwr.common import Context
 
 # #############################################################################
 # 1. Regular PyTorch pipeline: nn.Module, train, test, and DataLoader
@@ -122,7 +123,7 @@ def evaluate(self, parameters, config):
         return loss, len(testloader.dataset), {"accuracy": accuracy}
 
 
-def client_fn(cid: str):
+def client_fn(context: Context):
     """Create and return an instance of Flower `Client`."""
     return FlowerClient().to_client()
 

e2e/framework-fastai/client.py

@@ -5,7 +5,8 @@
 import torch
 from fastai.vision.all import *
 
-import flwr as fl
+from flwr.client import ClientApp, NumPyClient, start_client
+from flwr.common import Context
 
 warnings.filterwarnings("ignore", category=UserWarning)
 
@@ -29,7 +30,7 @@
 
 
 # Define Flower client
-class FlowerClient(fl.client.NumPyClient):
+class FlowerClient(NumPyClient):
     def get_parameters(self, config):
         return [val.cpu().numpy() for _, val in learn.model.state_dict().items()]
 
@@ -49,18 +50,18 @@ def evaluate(self, parameters, config):
         return loss, len(dls.valid), {"accuracy": 1 - error_rate}
 
 
-def client_fn(cid):
+def client_fn(context: Context):
     return FlowerClient().to_client()
 
 
-app = fl.client.ClientApp(
+app = ClientApp(
     client_fn=client_fn,
 )
 
 
 if __name__ == "__main__":
     # Start Flower client
-    fl.client.start_client(
+    start_client(
         server_address="127.0.0.1:8080",
         client=FlowerClient().to_client(),
     )

e2e/framework-jax/client.py

@@ -6,15 +6,16 @@
 import jax_training
 import numpy as np
 
-import flwr as fl
+from flwr.client import ClientApp, NumPyClient, start_client
+from flwr.common import Context
 
 # Load data and determine model shape
 train_x, train_y, test_x, test_y = jax_training.load_data()
 grad_fn = jax.grad(jax_training.loss_fn)
 model_shape = train_x.shape[1:]
 
 
-class FlowerClient(fl.client.NumPyClient):
+class FlowerClient(NumPyClient):
     def __init__(self):
         self.params = jax_training.load_model(model_shape)
 
@@ -48,16 +49,14 @@ def evaluate(
         return float(loss), num_examples, {"loss": float(loss)}
 
 
-def client_fn(cid):
+def client_fn(context: Context):
     return FlowerClient().to_client()
 
 
-app = fl.client.ClientApp(
+app = ClientApp(
     client_fn=client_fn,
 )
 
 if __name__ == "__main__":
     # Start Flower client
-    fl.client.start_client(
-        server_address="127.0.0.1:8080", client=FlowerClient().to_client()
-    )
+    start_client(server_address="127.0.0.1:8080", client=FlowerClient().to_client())

e2e/framework-opacus/client.py

@@ -9,7 +9,8 @@
 from torch.utils.data import DataLoader
 from torchvision.datasets import CIFAR10
 
-import flwr as fl
+from flwr.client import ClientApp, NumPyClient, start_client
+from flwr.common import Context
 
 # Define parameters.
 PARAMS = {
@@ -95,7 +96,7 @@ def load_data():
 
 
 # Define Flower client.
-class FlowerClient(fl.client.NumPyClient):
+class FlowerClient(NumPyClient):
     def __init__(self, model) -> None:
         super().__init__()
         # Create a privacy engine which will add DP and keep track of the privacy budget.
@@ -139,16 +140,16 @@ def evaluate(self, parameters, config):
         return float(loss), len(testloader), {"accuracy": float(accuracy)}
 
 
-def client_fn(cid):
+def client_fn(context: Context):
     model = Net()
     return FlowerClient(model).to_client()
 
 
-app = fl.client.ClientApp(
+app = ClientApp(
     client_fn=client_fn,
 )
 
 if __name__ == "__main__":
-    fl.client.start_client(
+    start_client(
         server_address="127.0.0.1:8080", client=FlowerClient(model).to_client()
     )

e2e/framework-pandas/client.py

@@ -3,7 +3,8 @@
 import numpy as np
 import pandas as pd
 
-import flwr as fl
+from flwr.client import ClientApp, NumPyClient, start_client
+from flwr.common import Context
 
 df = pd.read_csv("./data/client.csv")
 
@@ -16,7 +17,7 @@ def compute_hist(df: pd.DataFrame, col_name: str) -> np.ndarray:
 
 
 # Define Flower client
-class FlowerClient(fl.client.NumPyClient):
+class FlowerClient(NumPyClient):
     def fit(
         self, parameters: List[np.ndarray], config: Dict[str, str]
     ) -> Tuple[List[np.ndarray], int, Dict]:
@@ -32,17 +33,17 @@ def fit(
         )
 
 
-def client_fn(cid):
+def client_fn(context: Context):
     return FlowerClient().to_client()
 
 
-app = fl.client.ClientApp(
+app = ClientApp(
     client_fn=client_fn,
 )
 
 if __name__ == "__main__":
     # Start Flower client
-    fl.client.start_client(
+    start_client(
         server_address="127.0.0.1:8080",
         client=FlowerClient().to_client(),
     )

e2e/framework-pytorch-lightning/client.py

@@ -4,10 +4,11 @@
 import pytorch_lightning as pl
 import torch
 
-import flwr as fl
+from flwr.client import ClientApp, NumPyClient, start_client
+from flwr.common import Context
 
 
-class FlowerClient(fl.client.NumPyClient):
+class FlowerClient(NumPyClient):
     def __init__(self, model, train_loader, val_loader, test_loader):
         self.model = model
         self.train_loader = train_loader
@@ -51,15 +52,15 @@ def _set_parameters(model, parameters):
     model.load_state_dict(state_dict, strict=True)
 
 
-def client_fn(cid):
+def client_fn(context: Context):
     model = mnist.LitAutoEncoder()
     train_loader, val_loader, test_loader = mnist.load_data()
 
     # Flower client
     return FlowerClient(model, train_loader, val_loader, test_loader).to_client()
 
 
-app = fl.client.ClientApp(
+app = ClientApp(
     client_fn=client_fn,
 )
 
@@ -71,7 +72,7 @@ def main() -> None:
 
     # Flower client
     client = FlowerClient(model, train_loader, val_loader, test_loader).to_client()
-    fl.client.start_client(server_address="127.0.0.1:8080", client=client)
+    start_client(server_address="127.0.0.1:8080", client=client)
 
 
 if __name__ == "__main__":

e2e/framework-pytorch/client.py

@@ -10,8 +10,8 @@
 from torchvision.transforms import Compose, Normalize, ToTensor
 from tqdm import tqdm
 
-import flwr as fl
-from flwr.common import ConfigsRecord
+from flwr.client import ClientApp, NumPyClient, start_client
+from flwr.common import ConfigsRecord, Context
 
 # #############################################################################
 # 1. Regular PyTorch pipeline: nn.Module, train, test, and DataLoader
@@ -89,7 +89,7 @@ def load_data():
 
 
 # Define Flower client
-class FlowerClient(fl.client.NumPyClient):
+class FlowerClient(NumPyClient):
     def get_parameters(self, config):
         return [val.cpu().numpy() for _, val in net.state_dict().items()]
 
@@ -136,18 +136,18 @@ def set_parameters(model, parameters):
     return
 
 
-def client_fn(cid):
+def client_fn(context: Context):
     return FlowerClient().to_client()
 
 
-app = fl.client.ClientApp(
+app = ClientApp(
     client_fn=client_fn,
 )
 
 
 if __name__ == "__main__":
     # Start Flower client
-    fl.client.start_client(
+    start_client(
         server_address="127.0.0.1:8080",
         client=FlowerClient().to_client(),
     )