Fix vertical-fl example

examples/vertical-fl/pyproject.toml

@@ -3,14 +3,14 @@ requires = ["poetry-core>=1.4.0"]
 build-backend = "poetry.core.masonry.api"
 
 [tool.poetry]
-name = "quickstart-pytorch"
+name = "vertical-fl"
 version = "0.1.0"
-description = "PyTorch Federated Learning Quickstart with Flower"
+description = "PyTorch Vertical FL with Flower"
 authors = ["The Flower Authors <[email protected]>"]
 
 [tool.poetry.dependencies]
 python = ">=3.8,<3.11"
-flwr = ">=1.0,<2.0"
+flwr = { extras = ["simulation"], version = ">=1.0,<2.0" }
 torch = "2.1.0"
 matplotlib = "3.7.3"
 scikit-learn = "1.3.2"

examples/vertical-fl/requirements.txt

@@ -1,4 +1,4 @@
-flwr>=1.0, <2.0
+flwr[simulation]>=1.0, <2.0
 torch==2.1.0
 matplotlib==3.7.3
 scikit-learn==1.3.2

examples/vertical-fl/simulation.py

@@ -2,6 +2,7 @@
 import numpy as np
 from strategy import Strategy
 from client import FlowerClient
+from pathlib import Path
 from task import get_partitions_and_label
 
 partitions, label = get_partitions_and_label()
@@ -19,4 +20,6 @@ def client_fn(cid):
     strategy=Strategy(label),
 )
 
-np.save("_static/results/hist.npy", hist)
+results_dir = Path("_static/results")
+results_dir.mkdir(exist_ok=True)
+np.save(str(results_dir/"hist.npy"), hist)