flwr 1.24.0__py3-none-any.whl → 1.26.0__py3-none-any.whl

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
Files changed (204) hide show
  1. flwr/__init__.py +1 -1
  2. flwr/app/__init__.py +4 -1
  3. flwr/app/message_type.py +29 -0
  4. flwr/app/metadata.py +5 -2
  5. flwr/app/user_config.py +19 -0
  6. flwr/cli/app.py +37 -19
  7. flwr/cli/app_cmd/publish.py +25 -75
  8. flwr/cli/app_cmd/review.py +25 -66
  9. flwr/cli/auth_plugin/auth_plugin.py +5 -10
  10. flwr/cli/auth_plugin/noop_auth_plugin.py +1 -2
  11. flwr/cli/auth_plugin/oidc_cli_plugin.py +38 -38
  12. flwr/cli/build.py +15 -28
  13. flwr/cli/config/__init__.py +21 -0
  14. flwr/cli/config/ls.py +71 -0
  15. flwr/cli/config_migration.py +297 -0
  16. flwr/cli/config_utils.py +63 -156
  17. flwr/cli/constant.py +71 -0
  18. flwr/cli/federation/__init__.py +0 -2
  19. flwr/cli/federation/ls.py +256 -64
  20. flwr/cli/flower_config.py +429 -0
  21. flwr/cli/install.py +23 -62
  22. flwr/cli/log.py +23 -37
  23. flwr/cli/login/login.py +29 -63
  24. flwr/cli/ls.py +72 -61
  25. flwr/cli/new/new.py +98 -309
  26. flwr/cli/pull.py +19 -37
  27. flwr/cli/run/run.py +87 -100
  28. flwr/cli/run_utils.py +23 -5
  29. flwr/cli/stop.py +33 -74
  30. flwr/cli/supernode/ls.py +35 -62
  31. flwr/cli/supernode/register.py +31 -80
  32. flwr/cli/supernode/unregister.py +24 -70
  33. flwr/cli/typing.py +200 -0
  34. flwr/cli/utils.py +160 -412
  35. flwr/client/grpc_adapter_client/connection.py +2 -2
  36. flwr/client/grpc_rere_client/connection.py +9 -6
  37. flwr/client/grpc_rere_client/grpc_adapter.py +1 -1
  38. flwr/client/message_handler/message_handler.py +2 -1
  39. flwr/client/mod/centraldp_mods.py +1 -1
  40. flwr/client/mod/localdp_mod.py +1 -1
  41. flwr/client/mod/secure_aggregation/secaggplus_mod.py +1 -1
  42. flwr/client/rest_client/connection.py +6 -4
  43. flwr/client/run_info_store.py +2 -1
  44. flwr/clientapp/client_app.py +2 -1
  45. flwr/common/__init__.py +3 -2
  46. flwr/common/args.py +5 -5
  47. flwr/common/config.py +12 -17
  48. flwr/common/constant.py +3 -16
  49. flwr/common/context.py +2 -1
  50. flwr/common/exit/exit.py +4 -4
  51. flwr/common/exit/exit_code.py +6 -0
  52. flwr/common/grpc.py +2 -1
  53. flwr/common/logger.py +1 -1
  54. flwr/common/message.py +1 -1
  55. flwr/common/retry_invoker.py +13 -5
  56. flwr/common/secure_aggregation/ndarrays_arithmetic.py +5 -2
  57. flwr/common/serde.py +13 -5
  58. flwr/common/telemetry.py +1 -1
  59. flwr/common/typing.py +10 -3
  60. flwr/compat/client/app.py +6 -9
  61. flwr/compat/client/grpc_client/connection.py +2 -1
  62. flwr/compat/common/constant.py +29 -0
  63. flwr/compat/server/app.py +1 -1
  64. flwr/proto/clientappio_pb2.py +2 -2
  65. flwr/proto/clientappio_pb2_grpc.py +104 -88
  66. flwr/proto/clientappio_pb2_grpc.pyi +140 -80
  67. flwr/proto/federation_pb2.py +5 -3
  68. flwr/proto/federation_pb2.pyi +32 -2
  69. flwr/proto/fleet_pb2.py +10 -10
  70. flwr/proto/fleet_pb2.pyi +5 -1
  71. flwr/proto/run_pb2.py +18 -26
  72. flwr/proto/run_pb2.pyi +10 -58
  73. flwr/proto/serverappio_pb2.py +2 -2
  74. flwr/proto/serverappio_pb2_grpc.py +138 -207
  75. flwr/proto/serverappio_pb2_grpc.pyi +189 -155
  76. flwr/proto/simulationio_pb2.py +2 -2
  77. flwr/proto/simulationio_pb2_grpc.py +62 -90
  78. flwr/proto/simulationio_pb2_grpc.pyi +95 -55
  79. flwr/server/app.py +7 -13
  80. flwr/server/compat/grid_client_proxy.py +2 -1
  81. flwr/server/grid/grpc_grid.py +5 -5
  82. flwr/server/serverapp/app.py +11 -4
  83. flwr/server/superlink/fleet/grpc_adapter/grpc_adapter_servicer.py +1 -1
  84. flwr/server/superlink/fleet/grpc_rere/node_auth_server_interceptor.py +13 -12
  85. flwr/server/superlink/fleet/message_handler/message_handler.py +42 -2
  86. flwr/server/superlink/linkstate/__init__.py +2 -2
  87. flwr/server/superlink/linkstate/in_memory_linkstate.py +36 -10
  88. flwr/server/superlink/linkstate/linkstate.py +34 -21
  89. flwr/server/superlink/linkstate/linkstate_factory.py +16 -8
  90. flwr/server/superlink/linkstate/{sqlite_linkstate.py → sql_linkstate.py} +471 -516
  91. flwr/server/superlink/linkstate/utils.py +49 -2
  92. flwr/server/superlink/serverappio/serverappio_servicer.py +1 -33
  93. flwr/server/superlink/simulation/simulationio_servicer.py +0 -19
  94. flwr/server/utils/validator.py +1 -1
  95. flwr/server/workflow/default_workflows.py +2 -1
  96. flwr/server/workflow/secure_aggregation/secaggplus_workflow.py +1 -1
  97. flwr/serverapp/strategy/bulyan.py +7 -1
  98. flwr/serverapp/strategy/dp_fixed_clipping.py +9 -1
  99. flwr/serverapp/strategy/fedavg.py +1 -1
  100. flwr/serverapp/strategy/fedxgb_cyclic.py +1 -1
  101. flwr/simulation/ray_transport/ray_client_proxy.py +2 -6
  102. flwr/simulation/run_simulation.py +3 -12
  103. flwr/simulation/simulationio_connection.py +3 -3
  104. flwr/{common → supercore}/address.py +7 -33
  105. flwr/supercore/app_utils.py +2 -1
  106. flwr/supercore/constant.py +27 -2
  107. flwr/supercore/corestate/{sqlite_corestate.py → sql_corestate.py} +19 -23
  108. flwr/supercore/credential_store/__init__.py +33 -0
  109. flwr/supercore/credential_store/credential_store.py +34 -0
  110. flwr/supercore/credential_store/file_credential_store.py +76 -0
  111. flwr/{common → supercore}/date.py +0 -11
  112. flwr/supercore/ffs/disk_ffs.py +1 -1
  113. flwr/supercore/object_store/object_store_factory.py +14 -6
  114. flwr/supercore/object_store/{sqlite_object_store.py → sql_object_store.py} +115 -117
  115. flwr/supercore/sql_mixin.py +315 -0
  116. flwr/{cli/new/templates → supercore/state}/__init__.py +2 -2
  117. flwr/{cli/new/templates/app/code/flwr_tune → supercore/state/alembic}/__init__.py +2 -2
  118. flwr/supercore/state/alembic/env.py +103 -0
  119. flwr/supercore/state/alembic/script.py.mako +43 -0
  120. flwr/supercore/state/alembic/utils.py +239 -0
  121. flwr/{cli/new/templates/app → supercore/state/alembic/versions}/__init__.py +2 -2
  122. flwr/supercore/state/alembic/versions/rev_2026_01_28_initialize_migration_of_state_tables.py +200 -0
  123. flwr/supercore/state/schema/README.md +121 -0
  124. flwr/{cli/new/templates/app/code → supercore/state/schema}/__init__.py +2 -2
  125. flwr/supercore/state/schema/corestate_tables.py +36 -0
  126. flwr/supercore/state/schema/linkstate_tables.py +152 -0
  127. flwr/supercore/state/schema/objectstore_tables.py +90 -0
  128. flwr/supercore/superexec/run_superexec.py +2 -2
  129. flwr/supercore/utils.py +225 -0
  130. flwr/superlink/federation/federation_manager.py +2 -2
  131. flwr/superlink/federation/noop_federation_manager.py +8 -6
  132. flwr/superlink/servicer/control/control_grpc.py +2 -0
  133. flwr/superlink/servicer/control/control_servicer.py +106 -21
  134. flwr/supernode/cli/flower_supernode.py +2 -1
  135. flwr/supernode/nodestate/in_memory_nodestate.py +62 -1
  136. flwr/supernode/nodestate/nodestate.py +45 -0
  137. flwr/supernode/runtime/run_clientapp.py +14 -14
  138. flwr/supernode/servicer/clientappio/clientappio_servicer.py +13 -5
  139. flwr/supernode/start_client_internal.py +17 -10
  140. {flwr-1.24.0.dist-info → flwr-1.26.0.dist-info}/METADATA +8 -8
  141. {flwr-1.24.0.dist-info → flwr-1.26.0.dist-info}/RECORD +144 -184
  142. flwr/cli/federation/show.py +0 -317
  143. flwr/cli/new/templates/app/.gitignore.tpl +0 -163
  144. flwr/cli/new/templates/app/LICENSE.tpl +0 -202
  145. flwr/cli/new/templates/app/README.baseline.md.tpl +0 -127
  146. flwr/cli/new/templates/app/README.flowertune.md.tpl +0 -68
  147. flwr/cli/new/templates/app/README.md.tpl +0 -37
  148. flwr/cli/new/templates/app/code/__init__.baseline.py.tpl +0 -1
  149. flwr/cli/new/templates/app/code/__init__.py.tpl +0 -1
  150. flwr/cli/new/templates/app/code/__init__.pytorch_legacy_api.py.tpl +0 -1
  151. flwr/cli/new/templates/app/code/client.baseline.py.tpl +0 -75
  152. flwr/cli/new/templates/app/code/client.huggingface.py.tpl +0 -93
  153. flwr/cli/new/templates/app/code/client.jax.py.tpl +0 -71
  154. flwr/cli/new/templates/app/code/client.mlx.py.tpl +0 -102
  155. flwr/cli/new/templates/app/code/client.numpy.py.tpl +0 -46
  156. flwr/cli/new/templates/app/code/client.pytorch.py.tpl +0 -80
  157. flwr/cli/new/templates/app/code/client.pytorch_legacy_api.py.tpl +0 -55
  158. flwr/cli/new/templates/app/code/client.sklearn.py.tpl +0 -108
  159. flwr/cli/new/templates/app/code/client.tensorflow.py.tpl +0 -82
  160. flwr/cli/new/templates/app/code/client.xgboost.py.tpl +0 -110
  161. flwr/cli/new/templates/app/code/dataset.baseline.py.tpl +0 -36
  162. flwr/cli/new/templates/app/code/flwr_tune/client_app.py.tpl +0 -92
  163. flwr/cli/new/templates/app/code/flwr_tune/dataset.py.tpl +0 -87
  164. flwr/cli/new/templates/app/code/flwr_tune/models.py.tpl +0 -56
  165. flwr/cli/new/templates/app/code/flwr_tune/server_app.py.tpl +0 -73
  166. flwr/cli/new/templates/app/code/flwr_tune/strategy.py.tpl +0 -78
  167. flwr/cli/new/templates/app/code/model.baseline.py.tpl +0 -66
  168. flwr/cli/new/templates/app/code/server.baseline.py.tpl +0 -43
  169. flwr/cli/new/templates/app/code/server.huggingface.py.tpl +0 -42
  170. flwr/cli/new/templates/app/code/server.jax.py.tpl +0 -39
  171. flwr/cli/new/templates/app/code/server.mlx.py.tpl +0 -41
  172. flwr/cli/new/templates/app/code/server.numpy.py.tpl +0 -38
  173. flwr/cli/new/templates/app/code/server.pytorch.py.tpl +0 -41
  174. flwr/cli/new/templates/app/code/server.pytorch_legacy_api.py.tpl +0 -31
  175. flwr/cli/new/templates/app/code/server.sklearn.py.tpl +0 -44
  176. flwr/cli/new/templates/app/code/server.tensorflow.py.tpl +0 -38
  177. flwr/cli/new/templates/app/code/server.xgboost.py.tpl +0 -56
  178. flwr/cli/new/templates/app/code/strategy.baseline.py.tpl +0 -1
  179. flwr/cli/new/templates/app/code/task.huggingface.py.tpl +0 -98
  180. flwr/cli/new/templates/app/code/task.jax.py.tpl +0 -57
  181. flwr/cli/new/templates/app/code/task.mlx.py.tpl +0 -102
  182. flwr/cli/new/templates/app/code/task.numpy.py.tpl +0 -7
  183. flwr/cli/new/templates/app/code/task.pytorch.py.tpl +0 -99
  184. flwr/cli/new/templates/app/code/task.pytorch_legacy_api.py.tpl +0 -111
  185. flwr/cli/new/templates/app/code/task.sklearn.py.tpl +0 -67
  186. flwr/cli/new/templates/app/code/task.tensorflow.py.tpl +0 -52
  187. flwr/cli/new/templates/app/code/task.xgboost.py.tpl +0 -67
  188. flwr/cli/new/templates/app/code/utils.baseline.py.tpl +0 -1
  189. flwr/cli/new/templates/app/pyproject.baseline.toml.tpl +0 -146
  190. flwr/cli/new/templates/app/pyproject.flowertune.toml.tpl +0 -80
  191. flwr/cli/new/templates/app/pyproject.huggingface.toml.tpl +0 -65
  192. flwr/cli/new/templates/app/pyproject.jax.toml.tpl +0 -52
  193. flwr/cli/new/templates/app/pyproject.mlx.toml.tpl +0 -56
  194. flwr/cli/new/templates/app/pyproject.numpy.toml.tpl +0 -49
  195. flwr/cli/new/templates/app/pyproject.pytorch.toml.tpl +0 -53
  196. flwr/cli/new/templates/app/pyproject.pytorch_legacy_api.toml.tpl +0 -53
  197. flwr/cli/new/templates/app/pyproject.sklearn.toml.tpl +0 -52
  198. flwr/cli/new/templates/app/pyproject.tensorflow.toml.tpl +0 -53
  199. flwr/cli/new/templates/app/pyproject.xgboost.toml.tpl +0 -61
  200. flwr/common/pyproject.py +0 -42
  201. flwr/supercore/sqlite_mixin.py +0 -159
  202. /flwr/{common → supercore}/version.py +0 -0
  203. {flwr-1.24.0.dist-info → flwr-1.26.0.dist-info}/WHEEL +0 -0
  204. {flwr-1.24.0.dist-info → flwr-1.26.0.dist-info}/entry_points.txt +0 -0
@@ -1,67 +0,0 @@
1
- """$project_name: A Flower / $framework_str app."""
2
-
3
- import xgboost as xgb
4
- from flwr_datasets import FederatedDataset
5
- from flwr_datasets.partitioner import IidPartitioner
6
-
7
-
8
- def train_test_split(partition, test_fraction, seed):
9
- """Split the data into train and validation set given split rate."""
10
- train_test = partition.train_test_split(test_size=test_fraction, seed=seed)
11
- partition_train = train_test["train"]
12
- partition_test = train_test["test"]
13
-
14
- num_train = len(partition_train)
15
- num_test = len(partition_test)
16
-
17
- return partition_train, partition_test, num_train, num_test
18
-
19
-
20
- def transform_dataset_to_dmatrix(data):
21
- """Transform dataset to DMatrix format for xgboost."""
22
- x = data["inputs"]
23
- y = data["label"]
24
- new_data = xgb.DMatrix(x, label=y)
25
- return new_data
26
-
27
-
28
- fds = None # Cache FederatedDataset
29
-
30
-
31
- def load_data(partition_id, num_clients):
32
- """Load partition HIGGS data."""
33
- # Only initialize `FederatedDataset` once
34
- global fds
35
- if fds is None:
36
- partitioner = IidPartitioner(num_partitions=num_clients)
37
- fds = FederatedDataset(
38
- dataset="jxie/higgs",
39
- partitioners={"train": partitioner},
40
- )
41
-
42
- # Load the partition for this `partition_id`
43
- partition = fds.load_partition(partition_id, split="train")
44
- partition.set_format("numpy")
45
-
46
- # Train/test splitting
47
- train_data, valid_data, num_train, num_val = train_test_split(
48
- partition, test_fraction=0.2, seed=42
49
- )
50
-
51
- # Reformat data to DMatrix for xgboost
52
- train_dmatrix = transform_dataset_to_dmatrix(train_data)
53
- valid_dmatrix = transform_dataset_to_dmatrix(valid_data)
54
-
55
- return train_dmatrix, valid_dmatrix, num_train, num_val
56
-
57
-
58
- def replace_keys(input_dict, match="-", target="_"):
59
- """Recursively replace match string with target string in dictionary keys."""
60
- new_dict = {}
61
- for key, value in input_dict.items():
62
- new_key = key.replace(match, target)
63
- if isinstance(value, dict):
64
- new_dict[new_key] = replace_keys(value, match, target)
65
- else:
66
- new_dict[new_key] = value
67
- return new_dict
@@ -1 +0,0 @@
1
- """$project_name: A Flower Baseline."""
@@ -1,146 +0,0 @@
1
- # =====================================================================
2
- # For a full TOML configuration guide, check the Flower docs:
3
- # https://flower.ai/docs/framework/how-to-configure-pyproject-toml.html
4
- # =====================================================================
5
-
6
- [build-system]
7
- requires = ["hatchling"]
8
- build-backend = "hatchling.build"
9
-
10
- [project]
11
- name = "$package_name"
12
- version = "1.0.0"
13
- description = ""
14
- license = "Apache-2.0"
15
- # Dependencies for your Flower App
16
- dependencies = [
17
- "flwr[simulation]>=1.24.0",
18
- "flwr-datasets[vision]>=0.5.0",
19
- "torch==2.8.0",
20
- "torchvision==0.23.0",
21
- ]
22
-
23
- [tool.hatch.metadata]
24
- allow-direct-references = true
25
-
26
- [project.optional-dependencies]
27
- dev = [
28
- "isort==5.13.2",
29
- "black==24.2.0",
30
- "docformatter==1.7.5",
31
- "mypy==1.8.0",
32
- "pylint==3.3.1",
33
- "pytest==7.4.4",
34
- "pytest-watch==4.2.0",
35
- "ruff==0.4.5",
36
- "types-requests==2.31.0.20240125",
37
- ]
38
-
39
- [tool.isort]
40
- profile = "black"
41
-
42
- [tool.black]
43
- line-length = 88
44
- target-version = ["py310", "py311", "py312"]
45
-
46
- [tool.pytest.ini_options]
47
- minversion = "6.2"
48
- addopts = "-qq"
49
-
50
- [tool.mypy]
51
- ignore_missing_imports = true
52
- strict = false
53
- plugins = "numpy.typing.mypy_plugin"
54
-
55
- [tool.pylint."MESSAGES CONTROL"]
56
- disable = "duplicate-code,too-few-public-methods,useless-import-alias"
57
- good-names = "i,j,k,_,x,y,X,Y,K,N"
58
- max-args = 10
59
- max-attributes = 15
60
- max-locals = 36
61
- max-branches = 20
62
- max-statements = 55
63
-
64
- [tool.pylint.typecheck]
65
- generated-members = "numpy.*, torch.*, tensorflow.*"
66
-
67
- [[tool.mypy.overrides]]
68
- module = [
69
- "importlib.metadata.*",
70
- "importlib_metadata.*",
71
- ]
72
- follow_imports = "skip"
73
- follow_imports_for_stubs = true
74
- disallow_untyped_calls = false
75
-
76
- [[tool.mypy.overrides]]
77
- module = "torch.*"
78
- follow_imports = "skip"
79
- follow_imports_for_stubs = true
80
-
81
- [tool.docformatter]
82
- wrap-summaries = 88
83
- wrap-descriptions = 88
84
-
85
- [tool.ruff]
86
- target-version = "py310"
87
- line-length = 88
88
- exclude = [
89
- ".bzr",
90
- ".direnv",
91
- ".eggs",
92
- ".git",
93
- ".hg",
94
- ".mypy_cache",
95
- ".nox",
96
- ".pants.d",
97
- ".pytype",
98
- ".ruff_cache",
99
- ".svn",
100
- ".tox",
101
- ".venv",
102
- "__pypackages__",
103
- "_build",
104
- "buck-out",
105
- "build",
106
- "dist",
107
- "node_modules",
108
- "venv",
109
- "proto",
110
- ]
111
-
112
- [tool.ruff.lint]
113
- select = ["D", "E", "F", "W", "B", "ISC", "C4", "UP"]
114
- fixable = ["D", "E", "F", "W", "B", "ISC", "C4", "UP"]
115
- ignore = ["B024", "B027", "D205", "D209"]
116
-
117
- [tool.ruff.lint.pydocstyle]
118
- convention = "numpy"
119
-
120
- [tool.hatch.build.targets.wheel]
121
- packages = ["."]
122
-
123
- [tool.flwr.app]
124
- publisher = "$username"
125
-
126
- # Point to your ServerApp and ClientApp objects
127
- # Format: "<module>:<object>"
128
- [tool.flwr.app.components]
129
- serverapp = "$import_name.server_app:app"
130
- clientapp = "$import_name.client_app:app"
131
-
132
- # Custom config values accessible via `context.run_config`
133
- [tool.flwr.app.config]
134
- num-server-rounds = 3
135
- fraction-train = 0.5
136
- local-epochs = 1
137
-
138
- # Default federation to use when running the app
139
- [tool.flwr.federations]
140
- default = "local-simulation"
141
-
142
- # Local simulation federation with 10 virtual SuperNodes
143
- [tool.flwr.federations.local-simulation]
144
- options.num-supernodes = 10
145
- options.backend.client-resources.num-cpus = 2
146
- options.backend.client-resources.num-gpus = 0.0
@@ -1,80 +0,0 @@
1
- # =====================================================================
2
- # For a full TOML configuration guide, check the Flower docs:
3
- # https://flower.ai/docs/framework/how-to-configure-pyproject-toml.html
4
- # =====================================================================
5
-
6
- [build-system]
7
- requires = ["hatchling"]
8
- build-backend = "hatchling.build"
9
-
10
- [project]
11
- name = "$package_name"
12
- version = "1.0.0"
13
- description = ""
14
- license = "Apache-2.0"
15
- # Dependencies for your Flower App
16
- dependencies = [
17
- "flwr[simulation]>=1.24.0",
18
- "flwr-datasets>=0.5.0",
19
- "torch==2.4.0",
20
- "trl==0.8.1",
21
- "bitsandbytes==0.45.4",
22
- "scipy==1.13.0",
23
- "peft==0.6.2",
24
- "transformers==4.50.3",
25
- "sentencepiece==0.2.0",
26
- "omegaconf==2.3.0",
27
- "hf_transfer==0.1.8",
28
- ]
29
-
30
- [tool.hatch.build.targets.wheel]
31
- packages = ["."]
32
-
33
- [tool.flwr.app]
34
- publisher = "$username"
35
-
36
- # Point to your ServerApp and ClientApp objects
37
- # Format: "<module>:<object>"
38
- [tool.flwr.app.components]
39
- serverapp = "$import_name.server_app:app"
40
- clientapp = "$import_name.client_app:app"
41
-
42
- # Custom config values accessible via `context.run_config`
43
- [tool.flwr.app.config]
44
- model.name = "mistralai/Mistral-7B-v0.3"
45
- model.quantization = 4
46
- model.gradient-checkpointing = true
47
- model.lora.peft-lora-r = 32
48
- model.lora.peft-lora-alpha = 64
49
- train.save-every-round = 5
50
- train.learning-rate-max = 5e-5
51
- train.learning-rate-min = 1e-6
52
- train.seq-length = 512
53
- train.training-arguments.output-dir = ""
54
- train.training-arguments.learning-rate = ""
55
- train.training-arguments.per-device-train-batch-size = 16
56
- train.training-arguments.gradient-accumulation-steps = 1
57
- train.training-arguments.logging-steps = 10
58
- train.training-arguments.num-train-epochs = 3
59
- train.training-arguments.max-steps = 10
60
- train.training-arguments.save-steps = 1000
61
- train.training-arguments.save-total-limit = 10
62
- train.training-arguments.gradient-checkpointing = true
63
- train.training-arguments.lr-scheduler-type = "constant"
64
- strategy.fraction-train = $fraction_train
65
- strategy.fraction-evaluate = 0.0
66
- num-server-rounds = 200
67
-
68
- # Dataset config (static for FlowerTune LLM Leaderboard)
69
- [tool.flwr.app.config.static]
70
- dataset.name = "$dataset_name"
71
-
72
- # Default federation to use when running the app
73
- [tool.flwr.federations]
74
- default = "local-simulation"
75
-
76
- # Local simulation federation with $num_clients virtual SuperNodes
77
- [tool.flwr.federations.local-simulation]
78
- options.num-supernodes = $num_clients
79
- options.backend.client-resources.num-cpus = 6
80
- options.backend.client-resources.num-gpus = 1.0
@@ -1,65 +0,0 @@
1
- # =====================================================================
2
- # For a full TOML configuration guide, check the Flower docs:
3
- # https://flower.ai/docs/framework/how-to-configure-pyproject-toml.html
4
- # =====================================================================
5
-
6
- [build-system]
7
- requires = ["hatchling"]
8
- build-backend = "hatchling.build"
9
-
10
- [project]
11
- name = "$package_name"
12
- version = "1.0.0"
13
- description = ""
14
- license = "Apache-2.0"
15
- # Dependencies for your Flower App
16
- dependencies = [
17
- "flwr[simulation]>=1.24.0",
18
- "flwr-datasets>=0.5.0",
19
- "torch>=2.7.1",
20
- "transformers>=4.30.0,<5.0",
21
- "evaluate>=0.4.0,<1.0",
22
- "datasets>=2.0.0, <3.0",
23
- "scikit-learn>=1.6.1, <2.0",
24
- ]
25
-
26
- [tool.hatch.build.targets.wheel]
27
- packages = ["."]
28
-
29
- [tool.flwr.app]
30
- publisher = "$username"
31
-
32
- # Point to your ServerApp and ClientApp objects
33
- # Format: "<module>:<object>"
34
- [tool.flwr.app.components]
35
- serverapp = "$import_name.server_app:app"
36
- clientapp = "$import_name.client_app:app"
37
-
38
- # Custom config values accessible via `context.run_config`
39
- [tool.flwr.app.config]
40
- num-server-rounds = 3
41
- fraction-train = 0.5
42
- local-steps = 5
43
- model-name = "prajjwal1/bert-tiny" # Set a larger model if you have access to more GPU resources
44
- num-labels = 2
45
-
46
- # Default federation to use when running the app
47
- [tool.flwr.federations]
48
- default = "localhost"
49
-
50
- # Local simulation federation with 10 virtual SuperNodes
51
- [tool.flwr.federations.localhost]
52
- options.num-supernodes = 10
53
-
54
- # Local simulation federation with 10 virtual SuperNodes
55
- # making use of GPUs
56
- [tool.flwr.federations.localhost-gpu]
57
- options.num-supernodes = 10
58
- options.backend.client-resources.num-cpus = 4 # each ClientApp assumes to use 4CPUs
59
- options.backend.client-resources.num-gpus = 0.25 # at most 4 ClientApps will run in a given GPU
60
-
61
- # Remote federation example for use with SuperLink
62
- [tool.flwr.federations.remote-federation]
63
- address = "<SUPERLINK-ADDRESS>:<PORT>"
64
- insecure = true # Remove this line to enable TLS
65
- # root-certificates = "<PATH/TO/ca.crt>" # For TLS setup
@@ -1,52 +0,0 @@
1
- # =====================================================================
2
- # For a full TOML configuration guide, check the Flower docs:
3
- # https://flower.ai/docs/framework/how-to-configure-pyproject-toml.html
4
- # =====================================================================
5
-
6
- [build-system]
7
- requires = ["hatchling"]
8
- build-backend = "hatchling.build"
9
-
10
- [project]
11
- name = "$package_name"
12
- version = "1.0.0"
13
- description = ""
14
- license = "Apache-2.0"
15
- # Dependencies for your Flower App
16
- dependencies = [
17
- "flwr[simulation]>=1.24.0",
18
- "jax==0.4.30",
19
- "jaxlib==0.4.30",
20
- "scikit-learn==1.6.1",
21
- ]
22
-
23
- [tool.hatch.build.targets.wheel]
24
- packages = ["."]
25
-
26
- [tool.flwr.app]
27
- publisher = "$username"
28
-
29
- # Point to your ServerApp and ClientApp objects
30
- # Format: "<module>:<object>"
31
- [tool.flwr.app.components]
32
- serverapp = "$import_name.server_app:app"
33
- clientapp = "$import_name.client_app:app"
34
-
35
- # Custom config values accessible via `context.run_config`
36
- [tool.flwr.app.config]
37
- num-server-rounds = 3
38
- input-dim = 3
39
-
40
- # Default federation to use when running the app
41
- [tool.flwr.federations]
42
- default = "local-simulation"
43
-
44
- # Local simulation federation with 10 virtual SuperNodes
45
- [tool.flwr.federations.local-simulation]
46
- options.num-supernodes = 10
47
-
48
- # Remote federation example for use with SuperLink
49
- [tool.flwr.federations.remote-federation]
50
- address = "<SUPERLINK-ADDRESS>:<PORT>"
51
- insecure = true # Remove this line to enable TLS
52
- # root-certificates = "<PATH/TO/ca.crt>" # For TLS setup
@@ -1,56 +0,0 @@
1
- # =====================================================================
2
- # For a full TOML configuration guide, check the Flower docs:
3
- # https://flower.ai/docs/framework/how-to-configure-pyproject-toml.html
4
- # =====================================================================
5
-
6
- [build-system]
7
- requires = ["hatchling"]
8
- build-backend = "hatchling.build"
9
-
10
- [project]
11
- name = "$package_name"
12
- version = "1.0.0"
13
- description = ""
14
- license = "Apache-2.0"
15
- # Dependencies for your Flower App
16
- dependencies = [
17
- "flwr[simulation]>=1.24.0",
18
- "flwr-datasets[vision]>=0.5.0",
19
- "mlx==0.29.0",
20
- ]
21
-
22
- [tool.hatch.build.targets.wheel]
23
- packages = ["."]
24
-
25
- [tool.flwr.app]
26
- publisher = "$username"
27
-
28
- # Point to your ServerApp and ClientApp objects
29
- # Format: "<module>:<object>"
30
- [tool.flwr.app.components]
31
- serverapp = "$import_name.server_app:app"
32
- clientapp = "$import_name.client_app:app"
33
-
34
- # Custom config values accessible via `context.run_config`
35
- [tool.flwr.app.config]
36
- num-server-rounds = 3
37
- local-epochs = 1
38
- num-layers = 2
39
- input-dim = 784 # 28*28
40
- hidden-dim = 32
41
- batch-size = 256
42
- lr = 0.1
43
-
44
- # Default federation to use when running the app
45
- [tool.flwr.federations]
46
- default = "local-simulation"
47
-
48
- # Local simulation federation with 10 virtual SuperNodes
49
- [tool.flwr.federations.local-simulation]
50
- options.num-supernodes = 10
51
-
52
- # Remote federation example for use with SuperLink
53
- [tool.flwr.federations.remote-federation]
54
- address = "<SUPERLINK-ADDRESS>:<PORT>"
55
- insecure = true # Remove this line to enable TLS
56
- # root-certificates = "<PATH/TO/ca.crt>" # For TLS setup
@@ -1,49 +0,0 @@
1
- # =====================================================================
2
- # For a full TOML configuration guide, check the Flower docs:
3
- # https://flower.ai/docs/framework/how-to-configure-pyproject-toml.html
4
- # =====================================================================
5
-
6
- [build-system]
7
- requires = ["hatchling"]
8
- build-backend = "hatchling.build"
9
-
10
- [project]
11
- name = "$package_name"
12
- version = "1.0.0"
13
- description = ""
14
- license = "Apache-2.0"
15
- # Dependencies for your Flower App
16
- dependencies = [
17
- "flwr[simulation]>=1.24.0",
18
- "numpy>=2.0.2",
19
- ]
20
-
21
- [tool.hatch.build.targets.wheel]
22
- packages = ["."]
23
-
24
- [tool.flwr.app]
25
- publisher = "$username"
26
-
27
- # Point to your ServerApp and ClientApp objects
28
- # Format: "<module>:<object>"
29
- [tool.flwr.app.components]
30
- serverapp = "$import_name.server_app:app"
31
- clientapp = "$import_name.client_app:app"
32
-
33
- # Custom config values accessible via `context.run_config`
34
- [tool.flwr.app.config]
35
- num-server-rounds = 3
36
-
37
- # Default federation to use when running the app
38
- [tool.flwr.federations]
39
- default = "local-simulation"
40
-
41
- # Local simulation federation with 10 virtual SuperNodes
42
- [tool.flwr.federations.local-simulation]
43
- options.num-supernodes = 10
44
-
45
- # Remote federation example for use with SuperLink
46
- [tool.flwr.federations.remote-federation]
47
- address = "<SUPERLINK-ADDRESS>:<PORT>"
48
- insecure = true # Remove this line to enable TLS
49
- # root-certificates = "<PATH/TO/ca.crt>" # For TLS setup
@@ -1,53 +0,0 @@
1
- # =====================================================================
2
- # For a full TOML configuration guide, check the Flower docs:
3
- # https://flower.ai/docs/framework/how-to-configure-pyproject-toml.html
4
- # =====================================================================
5
-
6
- [build-system]
7
- requires = ["hatchling"]
8
- build-backend = "hatchling.build"
9
-
10
- [project]
11
- name = "$package_name"
12
- version = "1.0.0"
13
- description = ""
14
- license = "Apache-2.0"
15
- # Dependencies for your Flower App
16
- dependencies = [
17
- "flwr[simulation]>=1.24.0",
18
- "flwr-datasets[vision]>=0.5.0",
19
- "torch>=2.7.1",
20
- "torchvision>=0.22.1",
21
- ]
22
-
23
- [tool.hatch.build.targets.wheel]
24
- packages = ["."]
25
-
26
- [tool.flwr.app]
27
- publisher = "$username"
28
-
29
- # Point to your ServerApp and ClientApp objects
30
- [tool.flwr.app.components]
31
- serverapp = "$import_name.server_app:app"
32
- clientapp = "$import_name.client_app:app"
33
-
34
- # Custom config values accessible via `context.run_config`
35
- [tool.flwr.app.config]
36
- num-server-rounds = 3
37
- fraction-train = 0.5
38
- local-epochs = 1
39
- lr = 0.01
40
-
41
- # Default federation to use when running the app
42
- [tool.flwr.federations]
43
- default = "local-simulation"
44
-
45
- # Local simulation federation with 10 virtual SuperNodes
46
- [tool.flwr.federations.local-simulation]
47
- options.num-supernodes = 10
48
-
49
- # Remote federation example for use with SuperLink
50
- [tool.flwr.federations.remote-federation]
51
- address = "<SUPERLINK-ADDRESS>:<PORT>"
52
- insecure = true # Remove this line to enable TLS
53
- # root-certificates = "<PATH/TO/ca.crt>" # For TLS setup
@@ -1,53 +0,0 @@
1
- # =====================================================================
2
- # For a full TOML configuration guide, check the Flower docs:
3
- # https://flower.ai/docs/framework/how-to-configure-pyproject-toml.html
4
- # =====================================================================
5
-
6
- [build-system]
7
- requires = ["hatchling"]
8
- build-backend = "hatchling.build"
9
-
10
- [project]
11
- name = "$package_name"
12
- version = "1.0.0"
13
- description = ""
14
- license = "Apache-2.0"
15
- # Dependencies for your Flower App
16
- dependencies = [
17
- "flwr[simulation]>=1.24.0",
18
- "flwr-datasets[vision]>=0.5.0",
19
- "torch==2.7.1",
20
- "torchvision==0.22.1",
21
- ]
22
-
23
- [tool.hatch.build.targets.wheel]
24
- packages = ["."]
25
-
26
- [tool.flwr.app]
27
- publisher = "$username"
28
-
29
- # Point to your ServerApp and ClientApp objects
30
- # Format: "<module>:<object>"
31
- [tool.flwr.app.components]
32
- serverapp = "$import_name.server_app:app"
33
- clientapp = "$import_name.client_app:app"
34
-
35
- # Custom config values accessible via `context.run_config`
36
- [tool.flwr.app.config]
37
- num-server-rounds = 3
38
- fraction-fit = 0.5
39
- local-epochs = 1
40
-
41
- # Default federation to use when running the app
42
- [tool.flwr.federations]
43
- default = "local-simulation"
44
-
45
- # Local simulation federation with 10 virtual SuperNodes
46
- [tool.flwr.federations.local-simulation]
47
- options.num-supernodes = 10
48
-
49
- # Remote federation example for use with SuperLink
50
- [tool.flwr.federations.remote-federation]
51
- address = "<SUPERLINK-ADDRESS>:<PORT>"
52
- insecure = true # Remove this line to enable TLS
53
- # root-certificates = "<PATH/TO/ca.crt>" # For TLS setup
@@ -1,52 +0,0 @@
1
- # =====================================================================
2
- # For a full TOML configuration guide, check the Flower docs:
3
- # https://flower.ai/docs/framework/how-to-configure-pyproject-toml.html
4
- # =====================================================================
5
-
6
- [build-system]
7
- requires = ["hatchling"]
8
- build-backend = "hatchling.build"
9
-
10
- [project]
11
- name = "$package_name"
12
- version = "1.0.0"
13
- description = ""
14
- license = "Apache-2.0"
15
- # Dependencies for your Flower App
16
- dependencies = [
17
- "flwr[simulation]>=1.24.0",
18
- "flwr-datasets[vision]>=0.5.0",
19
- "scikit-learn>=1.6.1",
20
- ]
21
-
22
- [tool.hatch.build.targets.wheel]
23
- packages = ["."]
24
-
25
- [tool.flwr.app]
26
- publisher = "$username"
27
-
28
- # Point to your ServerApp and ClientApp objects
29
- # Format: "<module>:<object>"
30
- [tool.flwr.app.components]
31
- serverapp = "$import_name.server_app:app"
32
- clientapp = "$import_name.client_app:app"
33
-
34
- # Custom config values accessible via `context.run_config`
35
- [tool.flwr.app.config]
36
- num-server-rounds = 3
37
- penalty = "l2"
38
- local-epochs = 1
39
-
40
- # Default federation to use when running the app
41
- [tool.flwr.federations]
42
- default = "local-simulation"
43
-
44
- # Local simulation federation with 10 virtual SuperNodes
45
- [tool.flwr.federations.local-simulation]
46
- options.num-supernodes = 10
47
-
48
- # Remote federation example for use with SuperLink
49
- [tool.flwr.federations.remote-federation]
50
- address = "<SUPERLINK-ADDRESS>:<PORT>"
51
- insecure = true # Remove this line to enable TLS
52
- # root-certificates = "<PATH/TO/ca.crt>" # For TLS setup