capstone/pyproject.toml at main · matteovhaxt/capstone · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
# =====================================================================
# For a full TOML configuration guide, check the Flower docs:
# https://flower.ai/docs/framework/how-to-configure-pyproject-toml.html
# =====================================================================

[build-system]
requires = ["hatchling"]
build-backend = "hatchling.build"

[project]
name = "capstone"
version = "1.0.0"
description = ""
license = "Apache-2.0"
requires-python = ">=3.10,<3.11"
# Dependencies for your Flower App
dependencies = [
    "flwr[simulation]>=1.23.0",
    "flwr-datasets[vision]>=0.5.0",
    "torch==2.7.1",
    "torchvision==0.22.1",
    "ray[default]==2.31.0",
    "wandb==0.23.0",
    "dstack==0.19.40",
    "dstack[server]==0.19.40"
]

[project.optional-dependencies]
dev = [
    "ruff>=0.7.0,<0.8.0",
    "pytest>=8.3.0,<9.0.0",
    "pytest-mock>=3.14.0,<4.0.0",
]

[tool.hatch.build.targets.wheel]
packages = ["."]

[tool.flwr.app]
publisher = "flwrlabs"

# Point to your ServerApp and ClientApp objects
[tool.flwr.app.components]
serverapp = "src.server.server_app:app"
clientapp = "src.client.client_app:app"

# Custom config values accessible via `context.run_config`
[tool.flwr.app.config]
strategy = "auto"
num-server-rounds = 50
local-epochs = 2
lr = 0.001
batch-size = 64
num-clients-per-round = 10
round-timeout-seconds = 120
vps-host = ""
enable-run-service = false
enable-quantization = false
quantization-bits = 8
quantization-bits-split = ""
quantization-bits-seed = 0
simulated-latency-fp32-s = 1.6

# Default federation to use when running the app
[tool.flwr.federations]
default = "local-simulation"

# Local simulation federation with 10 virtual SuperNodes
[tool.flwr.federations.local-simulation]
options.num-supernodes = 10

# Remote federation example for use with SuperLink
[tool.flwr.federations.remote-federation]
address = "<SUPERLINK-ADDRESS>:<PORT>"
insecure = true  # Remove this line to enable TLS
# root-certificates = "<PATH/TO/ca.crt>"  # For TLS setup

[tool.ruff]
target-version = "py310"
line-length = 100

[tool.ruff.format]
quote-style = "double"
indent-style = "space"

[tool.ruff.lint]
select = [
    "E",  # pycodestyle errors
    "F",  # pyflakes
    "I",  # isort
    "B",  # flake8-bugbear
    "UP", # pyupgrade
    "SIM",# flake8-simplify
]