Fix mxnet quickstart example (#1574)

adap · Jan 14, 2023 · d40a654 · d40a654
1 parent cc40177
commit d40a654
Show file tree

Hide file tree

Showing 4 changed files with 5 additions and 3 deletions.
diff --git a/examples/quickstart_mxnet/.gitignore b/examples/quickstart_mxnet/.gitignore
@@ -0,0 +1 @@
+*.gz
diff --git a/examples/quickstart_mxnet/client.py b/examples/quickstart_mxnet/client.py
@@ -38,7 +38,7 @@ def model():
 
     # Flower Client
     class MNISTClient(fl.client.NumPyClient):
-        def get_parameters(self):
+        def get_parameters(self, config):
             param = []
             for val in model.collect_params(".*weight").values():
                 p = val.data()
@@ -54,7 +54,7 @@ def fit(self, parameters, config):
             self.set_parameters(parameters)
             [accuracy, loss], num_examples = train(model, train_data, epoch=2)
             results = {"accuracy": float(accuracy[1]), "loss": float(loss[1])}
-            return self.get_parameters(), num_examples, results
+            return self.get_parameters(config={}), num_examples, results
 
         def evaluate(self, parameters, config):
             self.set_parameters(parameters)

diff --git a/examples/quickstart_mxnet/pyproject.toml b/examples/quickstart_mxnet/pyproject.toml
@@ -13,3 +13,4 @@ python = "^3.7"
 flwr = "^0.17.0"
 # flwr = { path = "../../", develop = true }  # Development
 mxnet = "^1.7.0"
+numpy = "1.23.1"
diff --git a/examples/quickstart_mxnet/server.py b/examples/quickstart_mxnet/server.py
@@ -5,5 +5,5 @@
 if __name__ == "__main__":
     fl.server.start_server(
         server_address="0.0.0.0:8080",
-        config={"num_rounds": 3},
+        config=fl.server.ServerConfig(num_rounds=3),
     )