flwr-nightly 1.8.0.dev20240314__py3-none-any.whl → 1.15.0.dev20250114__py3-none-any.whl

Sign up to get free protection for your applications and to get access to all the features.
Files changed (311) hide show
  1. flwr/cli/app.py +16 -2
  2. flwr/cli/build.py +181 -0
  3. flwr/cli/cli_user_auth_interceptor.py +90 -0
  4. flwr/cli/config_utils.py +343 -0
  5. flwr/cli/example.py +4 -1
  6. flwr/cli/install.py +253 -0
  7. flwr/cli/log.py +182 -0
  8. flwr/{server/superlink/state → cli/login}/__init__.py +4 -10
  9. flwr/cli/login/login.py +88 -0
  10. flwr/cli/ls.py +327 -0
  11. flwr/cli/new/__init__.py +1 -0
  12. flwr/cli/new/new.py +210 -66
  13. flwr/cli/new/templates/app/.gitignore.tpl +163 -0
  14. flwr/cli/new/templates/app/LICENSE.tpl +202 -0
  15. flwr/cli/new/templates/app/README.baseline.md.tpl +127 -0
  16. flwr/cli/new/templates/app/README.flowertune.md.tpl +66 -0
  17. flwr/cli/new/templates/app/README.md.tpl +16 -32
  18. flwr/cli/new/templates/app/code/__init__.baseline.py.tpl +1 -0
  19. flwr/cli/new/templates/app/code/__init__.py.tpl +1 -1
  20. flwr/cli/new/templates/app/code/client.baseline.py.tpl +58 -0
  21. flwr/cli/new/templates/app/code/client.huggingface.py.tpl +55 -0
  22. flwr/cli/new/templates/app/code/client.jax.py.tpl +50 -0
  23. flwr/cli/new/templates/app/code/client.mlx.py.tpl +73 -0
  24. flwr/cli/new/templates/app/code/client.numpy.py.tpl +7 -7
  25. flwr/cli/new/templates/app/code/client.pytorch.py.tpl +30 -21
  26. flwr/cli/new/templates/app/code/client.sklearn.py.tpl +63 -0
  27. flwr/cli/new/templates/app/code/client.tensorflow.py.tpl +57 -1
  28. flwr/cli/new/templates/app/code/dataset.baseline.py.tpl +36 -0
  29. flwr/cli/new/templates/app/code/flwr_tune/__init__.py +15 -0
  30. flwr/cli/new/templates/app/code/flwr_tune/client_app.py.tpl +126 -0
  31. flwr/cli/new/templates/app/code/flwr_tune/dataset.py.tpl +87 -0
  32. flwr/cli/new/templates/app/code/flwr_tune/models.py.tpl +78 -0
  33. flwr/cli/new/templates/app/code/flwr_tune/server_app.py.tpl +94 -0
  34. flwr/cli/new/templates/app/code/flwr_tune/strategy.py.tpl +83 -0
  35. flwr/cli/new/templates/app/code/model.baseline.py.tpl +80 -0
  36. flwr/cli/new/templates/app/code/server.baseline.py.tpl +46 -0
  37. flwr/cli/new/templates/app/code/server.huggingface.py.tpl +38 -0
  38. flwr/cli/new/templates/app/code/server.jax.py.tpl +26 -0
  39. flwr/cli/new/templates/app/code/server.mlx.py.tpl +31 -0
  40. flwr/cli/new/templates/app/code/server.numpy.py.tpl +22 -9
  41. flwr/cli/new/templates/app/code/server.pytorch.py.tpl +21 -18
  42. flwr/cli/new/templates/app/code/server.sklearn.py.tpl +36 -0
  43. flwr/cli/new/templates/app/code/server.tensorflow.py.tpl +29 -1
  44. flwr/cli/new/templates/app/code/strategy.baseline.py.tpl +1 -0
  45. flwr/cli/new/templates/app/code/task.huggingface.py.tpl +102 -0
  46. flwr/cli/new/templates/app/code/task.jax.py.tpl +57 -0
  47. flwr/cli/new/templates/app/code/task.mlx.py.tpl +102 -0
  48. flwr/cli/new/templates/app/code/task.numpy.py.tpl +7 -0
  49. flwr/cli/new/templates/app/code/task.pytorch.py.tpl +29 -24
  50. flwr/cli/new/templates/app/code/task.sklearn.py.tpl +67 -0
  51. flwr/cli/new/templates/app/code/task.tensorflow.py.tpl +53 -0
  52. flwr/cli/new/templates/app/code/utils.baseline.py.tpl +1 -0
  53. flwr/cli/new/templates/app/pyproject.baseline.toml.tpl +138 -0
  54. flwr/cli/new/templates/app/pyproject.flowertune.toml.tpl +68 -0
  55. flwr/cli/new/templates/app/pyproject.huggingface.toml.tpl +46 -0
  56. flwr/cli/new/templates/app/pyproject.jax.toml.tpl +35 -0
  57. flwr/cli/new/templates/app/pyproject.mlx.toml.tpl +39 -0
  58. flwr/cli/new/templates/app/pyproject.numpy.toml.tpl +25 -12
  59. flwr/cli/new/templates/app/pyproject.pytorch.toml.tpl +29 -14
  60. flwr/cli/new/templates/app/pyproject.sklearn.toml.tpl +35 -0
  61. flwr/cli/new/templates/app/pyproject.tensorflow.toml.tpl +29 -14
  62. flwr/cli/run/__init__.py +1 -0
  63. flwr/cli/run/run.py +212 -34
  64. flwr/cli/stop.py +130 -0
  65. flwr/cli/utils.py +240 -5
  66. flwr/client/__init__.py +3 -2
  67. flwr/client/app.py +432 -255
  68. flwr/client/client.py +1 -11
  69. flwr/client/client_app.py +74 -13
  70. flwr/client/clientapp/__init__.py +22 -0
  71. flwr/client/clientapp/app.py +259 -0
  72. flwr/client/clientapp/clientappio_servicer.py +244 -0
  73. flwr/client/clientapp/utils.py +115 -0
  74. flwr/client/dpfedavg_numpy_client.py +7 -8
  75. flwr/client/grpc_adapter_client/__init__.py +15 -0
  76. flwr/client/grpc_adapter_client/connection.py +98 -0
  77. flwr/client/grpc_client/connection.py +21 -7
  78. flwr/client/grpc_rere_client/__init__.py +1 -1
  79. flwr/client/grpc_rere_client/client_interceptor.py +176 -0
  80. flwr/client/grpc_rere_client/connection.py +163 -56
  81. flwr/client/grpc_rere_client/grpc_adapter.py +167 -0
  82. flwr/client/heartbeat.py +74 -0
  83. flwr/client/message_handler/__init__.py +1 -1
  84. flwr/client/message_handler/message_handler.py +10 -11
  85. flwr/client/mod/__init__.py +5 -5
  86. flwr/client/mod/centraldp_mods.py +4 -2
  87. flwr/client/mod/comms_mods.py +5 -4
  88. flwr/client/mod/localdp_mod.py +10 -5
  89. flwr/client/mod/secure_aggregation/__init__.py +1 -1
  90. flwr/client/mod/secure_aggregation/secaggplus_mod.py +26 -26
  91. flwr/client/mod/utils.py +2 -4
  92. flwr/client/nodestate/__init__.py +26 -0
  93. flwr/client/nodestate/in_memory_nodestate.py +38 -0
  94. flwr/client/nodestate/nodestate.py +31 -0
  95. flwr/client/nodestate/nodestate_factory.py +38 -0
  96. flwr/client/numpy_client.py +8 -31
  97. flwr/client/rest_client/__init__.py +1 -1
  98. flwr/client/rest_client/connection.py +199 -176
  99. flwr/client/run_info_store.py +112 -0
  100. flwr/client/supernode/__init__.py +24 -0
  101. flwr/client/supernode/app.py +321 -0
  102. flwr/client/typing.py +1 -0
  103. flwr/common/__init__.py +17 -11
  104. flwr/common/address.py +47 -3
  105. flwr/common/args.py +153 -0
  106. flwr/common/auth_plugin/__init__.py +24 -0
  107. flwr/common/auth_plugin/auth_plugin.py +121 -0
  108. flwr/common/config.py +243 -0
  109. flwr/common/constant.py +132 -1
  110. flwr/common/context.py +32 -2
  111. flwr/common/date.py +22 -4
  112. flwr/common/differential_privacy.py +2 -2
  113. flwr/common/dp.py +2 -4
  114. flwr/common/exit_handlers.py +3 -3
  115. flwr/common/grpc.py +164 -5
  116. flwr/common/logger.py +230 -12
  117. flwr/common/message.py +191 -106
  118. flwr/common/object_ref.py +179 -44
  119. flwr/common/pyproject.py +1 -0
  120. flwr/common/record/__init__.py +2 -1
  121. flwr/common/record/configsrecord.py +58 -18
  122. flwr/common/record/metricsrecord.py +57 -17
  123. flwr/common/record/parametersrecord.py +88 -20
  124. flwr/common/record/recordset.py +153 -30
  125. flwr/common/record/typeddict.py +30 -55
  126. flwr/common/recordset_compat.py +31 -12
  127. flwr/common/retry_invoker.py +123 -30
  128. flwr/common/secure_aggregation/__init__.py +1 -1
  129. flwr/common/secure_aggregation/crypto/__init__.py +1 -1
  130. flwr/common/secure_aggregation/crypto/shamir.py +11 -11
  131. flwr/common/secure_aggregation/crypto/symmetric_encryption.py +68 -4
  132. flwr/common/secure_aggregation/ndarrays_arithmetic.py +17 -17
  133. flwr/common/secure_aggregation/quantization.py +8 -8
  134. flwr/common/secure_aggregation/secaggplus_constants.py +1 -1
  135. flwr/common/secure_aggregation/secaggplus_utils.py +10 -12
  136. flwr/common/serde.py +298 -19
  137. flwr/common/telemetry.py +65 -29
  138. flwr/common/typing.py +120 -19
  139. flwr/common/version.py +17 -3
  140. flwr/proto/clientappio_pb2.py +45 -0
  141. flwr/proto/clientappio_pb2.pyi +132 -0
  142. flwr/proto/clientappio_pb2_grpc.py +135 -0
  143. flwr/proto/clientappio_pb2_grpc.pyi +53 -0
  144. flwr/proto/exec_pb2.py +62 -0
  145. flwr/proto/exec_pb2.pyi +212 -0
  146. flwr/proto/exec_pb2_grpc.py +237 -0
  147. flwr/proto/exec_pb2_grpc.pyi +93 -0
  148. flwr/proto/fab_pb2.py +31 -0
  149. flwr/proto/fab_pb2.pyi +65 -0
  150. flwr/proto/fab_pb2_grpc.py +4 -0
  151. flwr/proto/fab_pb2_grpc.pyi +4 -0
  152. flwr/proto/fleet_pb2.py +42 -23
  153. flwr/proto/fleet_pb2.pyi +123 -1
  154. flwr/proto/fleet_pb2_grpc.py +170 -0
  155. flwr/proto/fleet_pb2_grpc.pyi +61 -0
  156. flwr/proto/grpcadapter_pb2.py +32 -0
  157. flwr/proto/grpcadapter_pb2.pyi +43 -0
  158. flwr/proto/grpcadapter_pb2_grpc.py +66 -0
  159. flwr/proto/grpcadapter_pb2_grpc.pyi +24 -0
  160. flwr/proto/log_pb2.py +29 -0
  161. flwr/proto/log_pb2.pyi +39 -0
  162. flwr/proto/log_pb2_grpc.py +4 -0
  163. flwr/proto/log_pb2_grpc.pyi +4 -0
  164. flwr/proto/message_pb2.py +41 -0
  165. flwr/proto/message_pb2.pyi +128 -0
  166. flwr/proto/message_pb2_grpc.py +4 -0
  167. flwr/proto/message_pb2_grpc.pyi +4 -0
  168. flwr/proto/node_pb2.py +1 -1
  169. flwr/proto/recordset_pb2.py +35 -33
  170. flwr/proto/recordset_pb2.pyi +40 -14
  171. flwr/proto/run_pb2.py +64 -0
  172. flwr/proto/run_pb2.pyi +268 -0
  173. flwr/proto/run_pb2_grpc.py +4 -0
  174. flwr/proto/run_pb2_grpc.pyi +4 -0
  175. flwr/proto/serverappio_pb2.py +52 -0
  176. flwr/proto/{driver_pb2.pyi → serverappio_pb2.pyi} +62 -20
  177. flwr/proto/serverappio_pb2_grpc.py +410 -0
  178. flwr/proto/serverappio_pb2_grpc.pyi +160 -0
  179. flwr/proto/simulationio_pb2.py +38 -0
  180. flwr/proto/simulationio_pb2.pyi +65 -0
  181. flwr/proto/simulationio_pb2_grpc.py +239 -0
  182. flwr/proto/simulationio_pb2_grpc.pyi +94 -0
  183. flwr/proto/task_pb2.py +7 -8
  184. flwr/proto/task_pb2.pyi +8 -5
  185. flwr/proto/transport_pb2.py +8 -8
  186. flwr/proto/transport_pb2.pyi +9 -6
  187. flwr/server/__init__.py +2 -10
  188. flwr/server/app.py +579 -402
  189. flwr/server/client_manager.py +8 -6
  190. flwr/server/compat/app.py +6 -62
  191. flwr/server/compat/app_utils.py +14 -8
  192. flwr/server/compat/driver_client_proxy.py +25 -58
  193. flwr/server/compat/legacy_context.py +5 -4
  194. flwr/server/driver/__init__.py +2 -0
  195. flwr/server/driver/driver.py +36 -131
  196. flwr/server/driver/grpc_driver.py +217 -81
  197. flwr/server/driver/inmemory_driver.py +182 -0
  198. flwr/server/history.py +28 -29
  199. flwr/server/run_serverapp.py +15 -126
  200. flwr/server/server.py +50 -44
  201. flwr/server/server_app.py +59 -10
  202. flwr/server/serverapp/__init__.py +22 -0
  203. flwr/server/serverapp/app.py +256 -0
  204. flwr/server/serverapp_components.py +52 -0
  205. flwr/server/strategy/__init__.py +2 -2
  206. flwr/server/strategy/aggregate.py +37 -23
  207. flwr/server/strategy/bulyan.py +9 -9
  208. flwr/server/strategy/dp_adaptive_clipping.py +25 -25
  209. flwr/server/strategy/dp_fixed_clipping.py +23 -22
  210. flwr/server/strategy/dpfedavg_adaptive.py +8 -8
  211. flwr/server/strategy/dpfedavg_fixed.py +13 -12
  212. flwr/server/strategy/fault_tolerant_fedavg.py +11 -11
  213. flwr/server/strategy/fedadagrad.py +9 -9
  214. flwr/server/strategy/fedadam.py +20 -10
  215. flwr/server/strategy/fedavg.py +16 -16
  216. flwr/server/strategy/fedavg_android.py +17 -17
  217. flwr/server/strategy/fedavgm.py +9 -9
  218. flwr/server/strategy/fedmedian.py +5 -5
  219. flwr/server/strategy/fedopt.py +6 -6
  220. flwr/server/strategy/fedprox.py +7 -7
  221. flwr/server/strategy/fedtrimmedavg.py +8 -8
  222. flwr/server/strategy/fedxgb_bagging.py +12 -12
  223. flwr/server/strategy/fedxgb_cyclic.py +10 -10
  224. flwr/server/strategy/fedxgb_nn_avg.py +6 -6
  225. flwr/server/strategy/fedyogi.py +9 -9
  226. flwr/server/strategy/krum.py +9 -9
  227. flwr/server/strategy/qfedavg.py +16 -16
  228. flwr/server/strategy/strategy.py +10 -10
  229. flwr/server/superlink/driver/__init__.py +2 -2
  230. flwr/server/superlink/driver/serverappio_grpc.py +61 -0
  231. flwr/server/superlink/driver/serverappio_servicer.py +363 -0
  232. flwr/server/superlink/ffs/__init__.py +24 -0
  233. flwr/server/superlink/ffs/disk_ffs.py +108 -0
  234. flwr/server/superlink/ffs/ffs.py +79 -0
  235. flwr/server/superlink/ffs/ffs_factory.py +47 -0
  236. flwr/server/superlink/fleet/__init__.py +1 -1
  237. flwr/server/superlink/fleet/grpc_adapter/__init__.py +15 -0
  238. flwr/server/superlink/fleet/grpc_adapter/grpc_adapter_servicer.py +162 -0
  239. flwr/server/superlink/fleet/grpc_bidi/__init__.py +1 -1
  240. flwr/server/superlink/fleet/grpc_bidi/flower_service_servicer.py +4 -2
  241. flwr/server/superlink/fleet/grpc_bidi/grpc_bridge.py +3 -2
  242. flwr/server/superlink/fleet/grpc_bidi/grpc_client_proxy.py +1 -1
  243. flwr/server/superlink/fleet/grpc_bidi/grpc_server.py +5 -154
  244. flwr/server/superlink/fleet/grpc_rere/__init__.py +1 -1
  245. flwr/server/superlink/fleet/grpc_rere/fleet_servicer.py +120 -13
  246. flwr/server/superlink/fleet/grpc_rere/server_interceptor.py +228 -0
  247. flwr/server/superlink/fleet/message_handler/__init__.py +1 -1
  248. flwr/server/superlink/fleet/message_handler/message_handler.py +153 -9
  249. flwr/server/superlink/fleet/rest_rere/__init__.py +1 -1
  250. flwr/server/superlink/fleet/rest_rere/rest_api.py +119 -81
  251. flwr/server/superlink/fleet/vce/__init__.py +1 -0
  252. flwr/server/superlink/fleet/vce/backend/__init__.py +4 -4
  253. flwr/server/superlink/fleet/vce/backend/backend.py +8 -9
  254. flwr/server/superlink/fleet/vce/backend/raybackend.py +87 -68
  255. flwr/server/superlink/fleet/vce/vce_api.py +208 -146
  256. flwr/server/superlink/linkstate/__init__.py +28 -0
  257. flwr/server/superlink/linkstate/in_memory_linkstate.py +581 -0
  258. flwr/server/superlink/linkstate/linkstate.py +389 -0
  259. flwr/server/superlink/{state/state_factory.py → linkstate/linkstate_factory.py} +19 -10
  260. flwr/server/superlink/linkstate/sqlite_linkstate.py +1236 -0
  261. flwr/server/superlink/linkstate/utils.py +389 -0
  262. flwr/server/superlink/simulation/__init__.py +15 -0
  263. flwr/server/superlink/simulation/simulationio_grpc.py +65 -0
  264. flwr/server/superlink/simulation/simulationio_servicer.py +186 -0
  265. flwr/server/superlink/utils.py +65 -0
  266. flwr/server/typing.py +2 -0
  267. flwr/server/utils/__init__.py +1 -1
  268. flwr/server/utils/tensorboard.py +5 -5
  269. flwr/server/utils/validator.py +31 -11
  270. flwr/server/workflow/default_workflows.py +70 -26
  271. flwr/server/workflow/secure_aggregation/secagg_workflow.py +1 -0
  272. flwr/server/workflow/secure_aggregation/secaggplus_workflow.py +40 -27
  273. flwr/simulation/__init__.py +12 -5
  274. flwr/simulation/app.py +247 -315
  275. flwr/simulation/legacy_app.py +402 -0
  276. flwr/simulation/ray_transport/__init__.py +1 -1
  277. flwr/simulation/ray_transport/ray_actor.py +42 -67
  278. flwr/simulation/ray_transport/ray_client_proxy.py +37 -17
  279. flwr/simulation/ray_transport/utils.py +1 -0
  280. flwr/simulation/run_simulation.py +306 -163
  281. flwr/simulation/simulationio_connection.py +89 -0
  282. flwr/superexec/__init__.py +15 -0
  283. flwr/superexec/app.py +59 -0
  284. flwr/superexec/deployment.py +188 -0
  285. flwr/superexec/exec_grpc.py +80 -0
  286. flwr/superexec/exec_servicer.py +231 -0
  287. flwr/superexec/exec_user_auth_interceptor.py +101 -0
  288. flwr/superexec/executor.py +96 -0
  289. flwr/superexec/simulation.py +124 -0
  290. {flwr_nightly-1.8.0.dev20240314.dist-info → flwr_nightly-1.15.0.dev20250114.dist-info}/METADATA +33 -26
  291. flwr_nightly-1.15.0.dev20250114.dist-info/RECORD +328 -0
  292. flwr_nightly-1.15.0.dev20250114.dist-info/entry_points.txt +12 -0
  293. flwr/cli/flower_toml.py +0 -140
  294. flwr/cli/new/templates/app/flower.toml.tpl +0 -13
  295. flwr/cli/new/templates/app/requirements.numpy.txt.tpl +0 -2
  296. flwr/cli/new/templates/app/requirements.pytorch.txt.tpl +0 -4
  297. flwr/cli/new/templates/app/requirements.tensorflow.txt.tpl +0 -4
  298. flwr/client/node_state.py +0 -48
  299. flwr/client/node_state_tests.py +0 -65
  300. flwr/proto/driver_pb2.py +0 -44
  301. flwr/proto/driver_pb2_grpc.py +0 -169
  302. flwr/proto/driver_pb2_grpc.pyi +0 -66
  303. flwr/server/superlink/driver/driver_grpc.py +0 -54
  304. flwr/server/superlink/driver/driver_servicer.py +0 -129
  305. flwr/server/superlink/state/in_memory_state.py +0 -230
  306. flwr/server/superlink/state/sqlite_state.py +0 -630
  307. flwr/server/superlink/state/state.py +0 -154
  308. flwr_nightly-1.8.0.dev20240314.dist-info/RECORD +0 -211
  309. flwr_nightly-1.8.0.dev20240314.dist-info/entry_points.txt +0 -9
  310. {flwr_nightly-1.8.0.dev20240314.dist-info → flwr_nightly-1.15.0.dev20250114.dist-info}/LICENSE +0 -0
  311. {flwr_nightly-1.8.0.dev20240314.dist-info → flwr_nightly-1.15.0.dev20250114.dist-info}/WHEEL +0 -0
@@ -0,0 +1,410 @@
1
+ # Generated by the gRPC Python protocol compiler plugin. DO NOT EDIT!
2
+ """Client and server classes corresponding to protobuf-defined services."""
3
+ import grpc
4
+
5
+ from flwr.proto import fab_pb2 as flwr_dot_proto_dot_fab__pb2
6
+ from flwr.proto import log_pb2 as flwr_dot_proto_dot_log__pb2
7
+ from flwr.proto import run_pb2 as flwr_dot_proto_dot_run__pb2
8
+ from flwr.proto import serverappio_pb2 as flwr_dot_proto_dot_serverappio__pb2
9
+
10
+
11
+ class ServerAppIoStub(object):
12
+ """Missing associated documentation comment in .proto file."""
13
+
14
+ def __init__(self, channel):
15
+ """Constructor.
16
+
17
+ Args:
18
+ channel: A grpc.Channel.
19
+ """
20
+ self.CreateRun = channel.unary_unary(
21
+ '/flwr.proto.ServerAppIo/CreateRun',
22
+ request_serializer=flwr_dot_proto_dot_run__pb2.CreateRunRequest.SerializeToString,
23
+ response_deserializer=flwr_dot_proto_dot_run__pb2.CreateRunResponse.FromString,
24
+ )
25
+ self.GetNodes = channel.unary_unary(
26
+ '/flwr.proto.ServerAppIo/GetNodes',
27
+ request_serializer=flwr_dot_proto_dot_serverappio__pb2.GetNodesRequest.SerializeToString,
28
+ response_deserializer=flwr_dot_proto_dot_serverappio__pb2.GetNodesResponse.FromString,
29
+ )
30
+ self.PushTaskIns = channel.unary_unary(
31
+ '/flwr.proto.ServerAppIo/PushTaskIns',
32
+ request_serializer=flwr_dot_proto_dot_serverappio__pb2.PushTaskInsRequest.SerializeToString,
33
+ response_deserializer=flwr_dot_proto_dot_serverappio__pb2.PushTaskInsResponse.FromString,
34
+ )
35
+ self.PullTaskRes = channel.unary_unary(
36
+ '/flwr.proto.ServerAppIo/PullTaskRes',
37
+ request_serializer=flwr_dot_proto_dot_serverappio__pb2.PullTaskResRequest.SerializeToString,
38
+ response_deserializer=flwr_dot_proto_dot_serverappio__pb2.PullTaskResResponse.FromString,
39
+ )
40
+ self.GetRun = channel.unary_unary(
41
+ '/flwr.proto.ServerAppIo/GetRun',
42
+ request_serializer=flwr_dot_proto_dot_run__pb2.GetRunRequest.SerializeToString,
43
+ response_deserializer=flwr_dot_proto_dot_run__pb2.GetRunResponse.FromString,
44
+ )
45
+ self.GetFab = channel.unary_unary(
46
+ '/flwr.proto.ServerAppIo/GetFab',
47
+ request_serializer=flwr_dot_proto_dot_fab__pb2.GetFabRequest.SerializeToString,
48
+ response_deserializer=flwr_dot_proto_dot_fab__pb2.GetFabResponse.FromString,
49
+ )
50
+ self.PullServerAppInputs = channel.unary_unary(
51
+ '/flwr.proto.ServerAppIo/PullServerAppInputs',
52
+ request_serializer=flwr_dot_proto_dot_serverappio__pb2.PullServerAppInputsRequest.SerializeToString,
53
+ response_deserializer=flwr_dot_proto_dot_serverappio__pb2.PullServerAppInputsResponse.FromString,
54
+ )
55
+ self.PushServerAppOutputs = channel.unary_unary(
56
+ '/flwr.proto.ServerAppIo/PushServerAppOutputs',
57
+ request_serializer=flwr_dot_proto_dot_serverappio__pb2.PushServerAppOutputsRequest.SerializeToString,
58
+ response_deserializer=flwr_dot_proto_dot_serverappio__pb2.PushServerAppOutputsResponse.FromString,
59
+ )
60
+ self.UpdateRunStatus = channel.unary_unary(
61
+ '/flwr.proto.ServerAppIo/UpdateRunStatus',
62
+ request_serializer=flwr_dot_proto_dot_run__pb2.UpdateRunStatusRequest.SerializeToString,
63
+ response_deserializer=flwr_dot_proto_dot_run__pb2.UpdateRunStatusResponse.FromString,
64
+ )
65
+ self.GetRunStatus = channel.unary_unary(
66
+ '/flwr.proto.ServerAppIo/GetRunStatus',
67
+ request_serializer=flwr_dot_proto_dot_run__pb2.GetRunStatusRequest.SerializeToString,
68
+ response_deserializer=flwr_dot_proto_dot_run__pb2.GetRunStatusResponse.FromString,
69
+ )
70
+ self.PushLogs = channel.unary_unary(
71
+ '/flwr.proto.ServerAppIo/PushLogs',
72
+ request_serializer=flwr_dot_proto_dot_log__pb2.PushLogsRequest.SerializeToString,
73
+ response_deserializer=flwr_dot_proto_dot_log__pb2.PushLogsResponse.FromString,
74
+ )
75
+
76
+
77
+ class ServerAppIoServicer(object):
78
+ """Missing associated documentation comment in .proto file."""
79
+
80
+ def CreateRun(self, request, context):
81
+ """Request run_id
82
+ """
83
+ context.set_code(grpc.StatusCode.UNIMPLEMENTED)
84
+ context.set_details('Method not implemented!')
85
+ raise NotImplementedError('Method not implemented!')
86
+
87
+ def GetNodes(self, request, context):
88
+ """Return a set of nodes
89
+ """
90
+ context.set_code(grpc.StatusCode.UNIMPLEMENTED)
91
+ context.set_details('Method not implemented!')
92
+ raise NotImplementedError('Method not implemented!')
93
+
94
+ def PushTaskIns(self, request, context):
95
+ """Create one or more tasks
96
+ """
97
+ context.set_code(grpc.StatusCode.UNIMPLEMENTED)
98
+ context.set_details('Method not implemented!')
99
+ raise NotImplementedError('Method not implemented!')
100
+
101
+ def PullTaskRes(self, request, context):
102
+ """Get task results
103
+ """
104
+ context.set_code(grpc.StatusCode.UNIMPLEMENTED)
105
+ context.set_details('Method not implemented!')
106
+ raise NotImplementedError('Method not implemented!')
107
+
108
+ def GetRun(self, request, context):
109
+ """Get run details
110
+ """
111
+ context.set_code(grpc.StatusCode.UNIMPLEMENTED)
112
+ context.set_details('Method not implemented!')
113
+ raise NotImplementedError('Method not implemented!')
114
+
115
+ def GetFab(self, request, context):
116
+ """Get FAB
117
+ """
118
+ context.set_code(grpc.StatusCode.UNIMPLEMENTED)
119
+ context.set_details('Method not implemented!')
120
+ raise NotImplementedError('Method not implemented!')
121
+
122
+ def PullServerAppInputs(self, request, context):
123
+ """Pull ServerApp inputs
124
+ """
125
+ context.set_code(grpc.StatusCode.UNIMPLEMENTED)
126
+ context.set_details('Method not implemented!')
127
+ raise NotImplementedError('Method not implemented!')
128
+
129
+ def PushServerAppOutputs(self, request, context):
130
+ """Push ServerApp outputs
131
+ """
132
+ context.set_code(grpc.StatusCode.UNIMPLEMENTED)
133
+ context.set_details('Method not implemented!')
134
+ raise NotImplementedError('Method not implemented!')
135
+
136
+ def UpdateRunStatus(self, request, context):
137
+ """Update the status of a given run
138
+ """
139
+ context.set_code(grpc.StatusCode.UNIMPLEMENTED)
140
+ context.set_details('Method not implemented!')
141
+ raise NotImplementedError('Method not implemented!')
142
+
143
+ def GetRunStatus(self, request, context):
144
+ """Get the status of a given run
145
+ """
146
+ context.set_code(grpc.StatusCode.UNIMPLEMENTED)
147
+ context.set_details('Method not implemented!')
148
+ raise NotImplementedError('Method not implemented!')
149
+
150
+ def PushLogs(self, request, context):
151
+ """Push ServerApp logs
152
+ """
153
+ context.set_code(grpc.StatusCode.UNIMPLEMENTED)
154
+ context.set_details('Method not implemented!')
155
+ raise NotImplementedError('Method not implemented!')
156
+
157
+
158
+ def add_ServerAppIoServicer_to_server(servicer, server):
159
+ rpc_method_handlers = {
160
+ 'CreateRun': grpc.unary_unary_rpc_method_handler(
161
+ servicer.CreateRun,
162
+ request_deserializer=flwr_dot_proto_dot_run__pb2.CreateRunRequest.FromString,
163
+ response_serializer=flwr_dot_proto_dot_run__pb2.CreateRunResponse.SerializeToString,
164
+ ),
165
+ 'GetNodes': grpc.unary_unary_rpc_method_handler(
166
+ servicer.GetNodes,
167
+ request_deserializer=flwr_dot_proto_dot_serverappio__pb2.GetNodesRequest.FromString,
168
+ response_serializer=flwr_dot_proto_dot_serverappio__pb2.GetNodesResponse.SerializeToString,
169
+ ),
170
+ 'PushTaskIns': grpc.unary_unary_rpc_method_handler(
171
+ servicer.PushTaskIns,
172
+ request_deserializer=flwr_dot_proto_dot_serverappio__pb2.PushTaskInsRequest.FromString,
173
+ response_serializer=flwr_dot_proto_dot_serverappio__pb2.PushTaskInsResponse.SerializeToString,
174
+ ),
175
+ 'PullTaskRes': grpc.unary_unary_rpc_method_handler(
176
+ servicer.PullTaskRes,
177
+ request_deserializer=flwr_dot_proto_dot_serverappio__pb2.PullTaskResRequest.FromString,
178
+ response_serializer=flwr_dot_proto_dot_serverappio__pb2.PullTaskResResponse.SerializeToString,
179
+ ),
180
+ 'GetRun': grpc.unary_unary_rpc_method_handler(
181
+ servicer.GetRun,
182
+ request_deserializer=flwr_dot_proto_dot_run__pb2.GetRunRequest.FromString,
183
+ response_serializer=flwr_dot_proto_dot_run__pb2.GetRunResponse.SerializeToString,
184
+ ),
185
+ 'GetFab': grpc.unary_unary_rpc_method_handler(
186
+ servicer.GetFab,
187
+ request_deserializer=flwr_dot_proto_dot_fab__pb2.GetFabRequest.FromString,
188
+ response_serializer=flwr_dot_proto_dot_fab__pb2.GetFabResponse.SerializeToString,
189
+ ),
190
+ 'PullServerAppInputs': grpc.unary_unary_rpc_method_handler(
191
+ servicer.PullServerAppInputs,
192
+ request_deserializer=flwr_dot_proto_dot_serverappio__pb2.PullServerAppInputsRequest.FromString,
193
+ response_serializer=flwr_dot_proto_dot_serverappio__pb2.PullServerAppInputsResponse.SerializeToString,
194
+ ),
195
+ 'PushServerAppOutputs': grpc.unary_unary_rpc_method_handler(
196
+ servicer.PushServerAppOutputs,
197
+ request_deserializer=flwr_dot_proto_dot_serverappio__pb2.PushServerAppOutputsRequest.FromString,
198
+ response_serializer=flwr_dot_proto_dot_serverappio__pb2.PushServerAppOutputsResponse.SerializeToString,
199
+ ),
200
+ 'UpdateRunStatus': grpc.unary_unary_rpc_method_handler(
201
+ servicer.UpdateRunStatus,
202
+ request_deserializer=flwr_dot_proto_dot_run__pb2.UpdateRunStatusRequest.FromString,
203
+ response_serializer=flwr_dot_proto_dot_run__pb2.UpdateRunStatusResponse.SerializeToString,
204
+ ),
205
+ 'GetRunStatus': grpc.unary_unary_rpc_method_handler(
206
+ servicer.GetRunStatus,
207
+ request_deserializer=flwr_dot_proto_dot_run__pb2.GetRunStatusRequest.FromString,
208
+ response_serializer=flwr_dot_proto_dot_run__pb2.GetRunStatusResponse.SerializeToString,
209
+ ),
210
+ 'PushLogs': grpc.unary_unary_rpc_method_handler(
211
+ servicer.PushLogs,
212
+ request_deserializer=flwr_dot_proto_dot_log__pb2.PushLogsRequest.FromString,
213
+ response_serializer=flwr_dot_proto_dot_log__pb2.PushLogsResponse.SerializeToString,
214
+ ),
215
+ }
216
+ generic_handler = grpc.method_handlers_generic_handler(
217
+ 'flwr.proto.ServerAppIo', rpc_method_handlers)
218
+ server.add_generic_rpc_handlers((generic_handler,))
219
+
220
+
221
+ # This class is part of an EXPERIMENTAL API.
222
+ class ServerAppIo(object):
223
+ """Missing associated documentation comment in .proto file."""
224
+
225
+ @staticmethod
226
+ def CreateRun(request,
227
+ target,
228
+ options=(),
229
+ channel_credentials=None,
230
+ call_credentials=None,
231
+ insecure=False,
232
+ compression=None,
233
+ wait_for_ready=None,
234
+ timeout=None,
235
+ metadata=None):
236
+ return grpc.experimental.unary_unary(request, target, '/flwr.proto.ServerAppIo/CreateRun',
237
+ flwr_dot_proto_dot_run__pb2.CreateRunRequest.SerializeToString,
238
+ flwr_dot_proto_dot_run__pb2.CreateRunResponse.FromString,
239
+ options, channel_credentials,
240
+ insecure, call_credentials, compression, wait_for_ready, timeout, metadata)
241
+
242
+ @staticmethod
243
+ def GetNodes(request,
244
+ target,
245
+ options=(),
246
+ channel_credentials=None,
247
+ call_credentials=None,
248
+ insecure=False,
249
+ compression=None,
250
+ wait_for_ready=None,
251
+ timeout=None,
252
+ metadata=None):
253
+ return grpc.experimental.unary_unary(request, target, '/flwr.proto.ServerAppIo/GetNodes',
254
+ flwr_dot_proto_dot_serverappio__pb2.GetNodesRequest.SerializeToString,
255
+ flwr_dot_proto_dot_serverappio__pb2.GetNodesResponse.FromString,
256
+ options, channel_credentials,
257
+ insecure, call_credentials, compression, wait_for_ready, timeout, metadata)
258
+
259
+ @staticmethod
260
+ def PushTaskIns(request,
261
+ target,
262
+ options=(),
263
+ channel_credentials=None,
264
+ call_credentials=None,
265
+ insecure=False,
266
+ compression=None,
267
+ wait_for_ready=None,
268
+ timeout=None,
269
+ metadata=None):
270
+ return grpc.experimental.unary_unary(request, target, '/flwr.proto.ServerAppIo/PushTaskIns',
271
+ flwr_dot_proto_dot_serverappio__pb2.PushTaskInsRequest.SerializeToString,
272
+ flwr_dot_proto_dot_serverappio__pb2.PushTaskInsResponse.FromString,
273
+ options, channel_credentials,
274
+ insecure, call_credentials, compression, wait_for_ready, timeout, metadata)
275
+
276
+ @staticmethod
277
+ def PullTaskRes(request,
278
+ target,
279
+ options=(),
280
+ channel_credentials=None,
281
+ call_credentials=None,
282
+ insecure=False,
283
+ compression=None,
284
+ wait_for_ready=None,
285
+ timeout=None,
286
+ metadata=None):
287
+ return grpc.experimental.unary_unary(request, target, '/flwr.proto.ServerAppIo/PullTaskRes',
288
+ flwr_dot_proto_dot_serverappio__pb2.PullTaskResRequest.SerializeToString,
289
+ flwr_dot_proto_dot_serverappio__pb2.PullTaskResResponse.FromString,
290
+ options, channel_credentials,
291
+ insecure, call_credentials, compression, wait_for_ready, timeout, metadata)
292
+
293
+ @staticmethod
294
+ def GetRun(request,
295
+ target,
296
+ options=(),
297
+ channel_credentials=None,
298
+ call_credentials=None,
299
+ insecure=False,
300
+ compression=None,
301
+ wait_for_ready=None,
302
+ timeout=None,
303
+ metadata=None):
304
+ return grpc.experimental.unary_unary(request, target, '/flwr.proto.ServerAppIo/GetRun',
305
+ flwr_dot_proto_dot_run__pb2.GetRunRequest.SerializeToString,
306
+ flwr_dot_proto_dot_run__pb2.GetRunResponse.FromString,
307
+ options, channel_credentials,
308
+ insecure, call_credentials, compression, wait_for_ready, timeout, metadata)
309
+
310
+ @staticmethod
311
+ def GetFab(request,
312
+ target,
313
+ options=(),
314
+ channel_credentials=None,
315
+ call_credentials=None,
316
+ insecure=False,
317
+ compression=None,
318
+ wait_for_ready=None,
319
+ timeout=None,
320
+ metadata=None):
321
+ return grpc.experimental.unary_unary(request, target, '/flwr.proto.ServerAppIo/GetFab',
322
+ flwr_dot_proto_dot_fab__pb2.GetFabRequest.SerializeToString,
323
+ flwr_dot_proto_dot_fab__pb2.GetFabResponse.FromString,
324
+ options, channel_credentials,
325
+ insecure, call_credentials, compression, wait_for_ready, timeout, metadata)
326
+
327
+ @staticmethod
328
+ def PullServerAppInputs(request,
329
+ target,
330
+ options=(),
331
+ channel_credentials=None,
332
+ call_credentials=None,
333
+ insecure=False,
334
+ compression=None,
335
+ wait_for_ready=None,
336
+ timeout=None,
337
+ metadata=None):
338
+ return grpc.experimental.unary_unary(request, target, '/flwr.proto.ServerAppIo/PullServerAppInputs',
339
+ flwr_dot_proto_dot_serverappio__pb2.PullServerAppInputsRequest.SerializeToString,
340
+ flwr_dot_proto_dot_serverappio__pb2.PullServerAppInputsResponse.FromString,
341
+ options, channel_credentials,
342
+ insecure, call_credentials, compression, wait_for_ready, timeout, metadata)
343
+
344
+ @staticmethod
345
+ def PushServerAppOutputs(request,
346
+ target,
347
+ options=(),
348
+ channel_credentials=None,
349
+ call_credentials=None,
350
+ insecure=False,
351
+ compression=None,
352
+ wait_for_ready=None,
353
+ timeout=None,
354
+ metadata=None):
355
+ return grpc.experimental.unary_unary(request, target, '/flwr.proto.ServerAppIo/PushServerAppOutputs',
356
+ flwr_dot_proto_dot_serverappio__pb2.PushServerAppOutputsRequest.SerializeToString,
357
+ flwr_dot_proto_dot_serverappio__pb2.PushServerAppOutputsResponse.FromString,
358
+ options, channel_credentials,
359
+ insecure, call_credentials, compression, wait_for_ready, timeout, metadata)
360
+
361
+ @staticmethod
362
+ def UpdateRunStatus(request,
363
+ target,
364
+ options=(),
365
+ channel_credentials=None,
366
+ call_credentials=None,
367
+ insecure=False,
368
+ compression=None,
369
+ wait_for_ready=None,
370
+ timeout=None,
371
+ metadata=None):
372
+ return grpc.experimental.unary_unary(request, target, '/flwr.proto.ServerAppIo/UpdateRunStatus',
373
+ flwr_dot_proto_dot_run__pb2.UpdateRunStatusRequest.SerializeToString,
374
+ flwr_dot_proto_dot_run__pb2.UpdateRunStatusResponse.FromString,
375
+ options, channel_credentials,
376
+ insecure, call_credentials, compression, wait_for_ready, timeout, metadata)
377
+
378
+ @staticmethod
379
+ def GetRunStatus(request,
380
+ target,
381
+ options=(),
382
+ channel_credentials=None,
383
+ call_credentials=None,
384
+ insecure=False,
385
+ compression=None,
386
+ wait_for_ready=None,
387
+ timeout=None,
388
+ metadata=None):
389
+ return grpc.experimental.unary_unary(request, target, '/flwr.proto.ServerAppIo/GetRunStatus',
390
+ flwr_dot_proto_dot_run__pb2.GetRunStatusRequest.SerializeToString,
391
+ flwr_dot_proto_dot_run__pb2.GetRunStatusResponse.FromString,
392
+ options, channel_credentials,
393
+ insecure, call_credentials, compression, wait_for_ready, timeout, metadata)
394
+
395
+ @staticmethod
396
+ def PushLogs(request,
397
+ target,
398
+ options=(),
399
+ channel_credentials=None,
400
+ call_credentials=None,
401
+ insecure=False,
402
+ compression=None,
403
+ wait_for_ready=None,
404
+ timeout=None,
405
+ metadata=None):
406
+ return grpc.experimental.unary_unary(request, target, '/flwr.proto.ServerAppIo/PushLogs',
407
+ flwr_dot_proto_dot_log__pb2.PushLogsRequest.SerializeToString,
408
+ flwr_dot_proto_dot_log__pb2.PushLogsResponse.FromString,
409
+ options, channel_credentials,
410
+ insecure, call_credentials, compression, wait_for_ready, timeout, metadata)
@@ -0,0 +1,160 @@
1
+ """
2
+ @generated by mypy-protobuf. Do not edit manually!
3
+ isort:skip_file
4
+ """
5
+ import abc
6
+ import flwr.proto.fab_pb2
7
+ import flwr.proto.log_pb2
8
+ import flwr.proto.run_pb2
9
+ import flwr.proto.serverappio_pb2
10
+ import grpc
11
+
12
+ class ServerAppIoStub:
13
+ def __init__(self, channel: grpc.Channel) -> None: ...
14
+ CreateRun: grpc.UnaryUnaryMultiCallable[
15
+ flwr.proto.run_pb2.CreateRunRequest,
16
+ flwr.proto.run_pb2.CreateRunResponse]
17
+ """Request run_id"""
18
+
19
+ GetNodes: grpc.UnaryUnaryMultiCallable[
20
+ flwr.proto.serverappio_pb2.GetNodesRequest,
21
+ flwr.proto.serverappio_pb2.GetNodesResponse]
22
+ """Return a set of nodes"""
23
+
24
+ PushTaskIns: grpc.UnaryUnaryMultiCallable[
25
+ flwr.proto.serverappio_pb2.PushTaskInsRequest,
26
+ flwr.proto.serverappio_pb2.PushTaskInsResponse]
27
+ """Create one or more tasks"""
28
+
29
+ PullTaskRes: grpc.UnaryUnaryMultiCallable[
30
+ flwr.proto.serverappio_pb2.PullTaskResRequest,
31
+ flwr.proto.serverappio_pb2.PullTaskResResponse]
32
+ """Get task results"""
33
+
34
+ GetRun: grpc.UnaryUnaryMultiCallable[
35
+ flwr.proto.run_pb2.GetRunRequest,
36
+ flwr.proto.run_pb2.GetRunResponse]
37
+ """Get run details"""
38
+
39
+ GetFab: grpc.UnaryUnaryMultiCallable[
40
+ flwr.proto.fab_pb2.GetFabRequest,
41
+ flwr.proto.fab_pb2.GetFabResponse]
42
+ """Get FAB"""
43
+
44
+ PullServerAppInputs: grpc.UnaryUnaryMultiCallable[
45
+ flwr.proto.serverappio_pb2.PullServerAppInputsRequest,
46
+ flwr.proto.serverappio_pb2.PullServerAppInputsResponse]
47
+ """Pull ServerApp inputs"""
48
+
49
+ PushServerAppOutputs: grpc.UnaryUnaryMultiCallable[
50
+ flwr.proto.serverappio_pb2.PushServerAppOutputsRequest,
51
+ flwr.proto.serverappio_pb2.PushServerAppOutputsResponse]
52
+ """Push ServerApp outputs"""
53
+
54
+ UpdateRunStatus: grpc.UnaryUnaryMultiCallable[
55
+ flwr.proto.run_pb2.UpdateRunStatusRequest,
56
+ flwr.proto.run_pb2.UpdateRunStatusResponse]
57
+ """Update the status of a given run"""
58
+
59
+ GetRunStatus: grpc.UnaryUnaryMultiCallable[
60
+ flwr.proto.run_pb2.GetRunStatusRequest,
61
+ flwr.proto.run_pb2.GetRunStatusResponse]
62
+ """Get the status of a given run"""
63
+
64
+ PushLogs: grpc.UnaryUnaryMultiCallable[
65
+ flwr.proto.log_pb2.PushLogsRequest,
66
+ flwr.proto.log_pb2.PushLogsResponse]
67
+ """Push ServerApp logs"""
68
+
69
+
70
+ class ServerAppIoServicer(metaclass=abc.ABCMeta):
71
+ @abc.abstractmethod
72
+ def CreateRun(self,
73
+ request: flwr.proto.run_pb2.CreateRunRequest,
74
+ context: grpc.ServicerContext,
75
+ ) -> flwr.proto.run_pb2.CreateRunResponse:
76
+ """Request run_id"""
77
+ pass
78
+
79
+ @abc.abstractmethod
80
+ def GetNodes(self,
81
+ request: flwr.proto.serverappio_pb2.GetNodesRequest,
82
+ context: grpc.ServicerContext,
83
+ ) -> flwr.proto.serverappio_pb2.GetNodesResponse:
84
+ """Return a set of nodes"""
85
+ pass
86
+
87
+ @abc.abstractmethod
88
+ def PushTaskIns(self,
89
+ request: flwr.proto.serverappio_pb2.PushTaskInsRequest,
90
+ context: grpc.ServicerContext,
91
+ ) -> flwr.proto.serverappio_pb2.PushTaskInsResponse:
92
+ """Create one or more tasks"""
93
+ pass
94
+
95
+ @abc.abstractmethod
96
+ def PullTaskRes(self,
97
+ request: flwr.proto.serverappio_pb2.PullTaskResRequest,
98
+ context: grpc.ServicerContext,
99
+ ) -> flwr.proto.serverappio_pb2.PullTaskResResponse:
100
+ """Get task results"""
101
+ pass
102
+
103
+ @abc.abstractmethod
104
+ def GetRun(self,
105
+ request: flwr.proto.run_pb2.GetRunRequest,
106
+ context: grpc.ServicerContext,
107
+ ) -> flwr.proto.run_pb2.GetRunResponse:
108
+ """Get run details"""
109
+ pass
110
+
111
+ @abc.abstractmethod
112
+ def GetFab(self,
113
+ request: flwr.proto.fab_pb2.GetFabRequest,
114
+ context: grpc.ServicerContext,
115
+ ) -> flwr.proto.fab_pb2.GetFabResponse:
116
+ """Get FAB"""
117
+ pass
118
+
119
+ @abc.abstractmethod
120
+ def PullServerAppInputs(self,
121
+ request: flwr.proto.serverappio_pb2.PullServerAppInputsRequest,
122
+ context: grpc.ServicerContext,
123
+ ) -> flwr.proto.serverappio_pb2.PullServerAppInputsResponse:
124
+ """Pull ServerApp inputs"""
125
+ pass
126
+
127
+ @abc.abstractmethod
128
+ def PushServerAppOutputs(self,
129
+ request: flwr.proto.serverappio_pb2.PushServerAppOutputsRequest,
130
+ context: grpc.ServicerContext,
131
+ ) -> flwr.proto.serverappio_pb2.PushServerAppOutputsResponse:
132
+ """Push ServerApp outputs"""
133
+ pass
134
+
135
+ @abc.abstractmethod
136
+ def UpdateRunStatus(self,
137
+ request: flwr.proto.run_pb2.UpdateRunStatusRequest,
138
+ context: grpc.ServicerContext,
139
+ ) -> flwr.proto.run_pb2.UpdateRunStatusResponse:
140
+ """Update the status of a given run"""
141
+ pass
142
+
143
+ @abc.abstractmethod
144
+ def GetRunStatus(self,
145
+ request: flwr.proto.run_pb2.GetRunStatusRequest,
146
+ context: grpc.ServicerContext,
147
+ ) -> flwr.proto.run_pb2.GetRunStatusResponse:
148
+ """Get the status of a given run"""
149
+ pass
150
+
151
+ @abc.abstractmethod
152
+ def PushLogs(self,
153
+ request: flwr.proto.log_pb2.PushLogsRequest,
154
+ context: grpc.ServicerContext,
155
+ ) -> flwr.proto.log_pb2.PushLogsResponse:
156
+ """Push ServerApp logs"""
157
+ pass
158
+
159
+
160
+ def add_ServerAppIoServicer_to_server(servicer: ServerAppIoServicer, server: grpc.Server) -> None: ...
@@ -0,0 +1,38 @@
1
+ # -*- coding: utf-8 -*-
2
+ # Generated by the protocol buffer compiler. DO NOT EDIT!
3
+ # source: flwr/proto/simulationio.proto
4
+ # Protobuf Python Version: 4.25.0
5
+ """Generated protocol buffer code."""
6
+ from google.protobuf import descriptor as _descriptor
7
+ from google.protobuf import descriptor_pool as _descriptor_pool
8
+ from google.protobuf import symbol_database as _symbol_database
9
+ from google.protobuf.internal import builder as _builder
10
+ # @@protoc_insertion_point(imports)
11
+
12
+ _sym_db = _symbol_database.Default()
13
+
14
+
15
+ from flwr.proto import log_pb2 as flwr_dot_proto_dot_log__pb2
16
+ from flwr.proto import message_pb2 as flwr_dot_proto_dot_message__pb2
17
+ from flwr.proto import run_pb2 as flwr_dot_proto_dot_run__pb2
18
+ from flwr.proto import fab_pb2 as flwr_dot_proto_dot_fab__pb2
19
+
20
+
21
+ DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n\x1d\x66lwr/proto/simulationio.proto\x12\nflwr.proto\x1a\x14\x66lwr/proto/log.proto\x1a\x18\x66lwr/proto/message.proto\x1a\x14\x66lwr/proto/run.proto\x1a\x14\x66lwr/proto/fab.proto\"\x1d\n\x1bPullSimulationInputsRequest\"\x80\x01\n\x1cPullSimulationInputsResponse\x12$\n\x07\x63ontext\x18\x01 \x01(\x0b\x32\x13.flwr.proto.Context\x12\x1c\n\x03run\x18\x02 \x01(\x0b\x32\x0f.flwr.proto.Run\x12\x1c\n\x03\x66\x61\x62\x18\x03 \x01(\x0b\x32\x0f.flwr.proto.Fab\"T\n\x1cPushSimulationOutputsRequest\x12\x0e\n\x06run_id\x18\x01 \x01(\x04\x12$\n\x07\x63ontext\x18\x02 \x01(\x0b\x32\x13.flwr.proto.Context\"\x1f\n\x1dPushSimulationOutputsResponse2\xd4\x04\n\x0cSimulationIo\x12k\n\x14PullSimulationInputs\x12\'.flwr.proto.PullSimulationInputsRequest\x1a(.flwr.proto.PullSimulationInputsResponse\"\x00\x12n\n\x15PushSimulationOutputs\x12(.flwr.proto.PushSimulationOutputsRequest\x1a).flwr.proto.PushSimulationOutputsResponse\"\x00\x12\\\n\x0fUpdateRunStatus\x12\".flwr.proto.UpdateRunStatusRequest\x1a#.flwr.proto.UpdateRunStatusResponse\"\x00\x12G\n\x08PushLogs\x12\x1b.flwr.proto.PushLogsRequest\x1a\x1c.flwr.proto.PushLogsResponse\"\x00\x12k\n\x14GetFederationOptions\x12\'.flwr.proto.GetFederationOptionsRequest\x1a(.flwr.proto.GetFederationOptionsResponse\"\x00\x12S\n\x0cGetRunStatus\x12\x1f.flwr.proto.GetRunStatusRequest\x1a .flwr.proto.GetRunStatusResponse\"\x00\x62\x06proto3')
22
+
23
+ _globals = globals()
24
+ _builder.BuildMessageAndEnumDescriptors(DESCRIPTOR, _globals)
25
+ _builder.BuildTopDescriptorsAndMessages(DESCRIPTOR, 'flwr.proto.simulationio_pb2', _globals)
26
+ if _descriptor._USE_C_DESCRIPTORS == False:
27
+ DESCRIPTOR._options = None
28
+ _globals['_PULLSIMULATIONINPUTSREQUEST']._serialized_start=137
29
+ _globals['_PULLSIMULATIONINPUTSREQUEST']._serialized_end=166
30
+ _globals['_PULLSIMULATIONINPUTSRESPONSE']._serialized_start=169
31
+ _globals['_PULLSIMULATIONINPUTSRESPONSE']._serialized_end=297
32
+ _globals['_PUSHSIMULATIONOUTPUTSREQUEST']._serialized_start=299
33
+ _globals['_PUSHSIMULATIONOUTPUTSREQUEST']._serialized_end=383
34
+ _globals['_PUSHSIMULATIONOUTPUTSRESPONSE']._serialized_start=385
35
+ _globals['_PUSHSIMULATIONOUTPUTSRESPONSE']._serialized_end=416
36
+ _globals['_SIMULATIONIO']._serialized_start=419
37
+ _globals['_SIMULATIONIO']._serialized_end=1015
38
+ # @@protoc_insertion_point(module_scope)
@@ -0,0 +1,65 @@
1
+ """
2
+ @generated by mypy-protobuf. Do not edit manually!
3
+ isort:skip_file
4
+ """
5
+ import builtins
6
+ import flwr.proto.fab_pb2
7
+ import flwr.proto.message_pb2
8
+ import flwr.proto.run_pb2
9
+ import google.protobuf.descriptor
10
+ import google.protobuf.message
11
+ import typing
12
+ import typing_extensions
13
+
14
+ DESCRIPTOR: google.protobuf.descriptor.FileDescriptor
15
+
16
+ class PullSimulationInputsRequest(google.protobuf.message.Message):
17
+ """PullSimulationInputs messages"""
18
+ DESCRIPTOR: google.protobuf.descriptor.Descriptor
19
+ def __init__(self,
20
+ ) -> None: ...
21
+ global___PullSimulationInputsRequest = PullSimulationInputsRequest
22
+
23
+ class PullSimulationInputsResponse(google.protobuf.message.Message):
24
+ DESCRIPTOR: google.protobuf.descriptor.Descriptor
25
+ CONTEXT_FIELD_NUMBER: builtins.int
26
+ RUN_FIELD_NUMBER: builtins.int
27
+ FAB_FIELD_NUMBER: builtins.int
28
+ @property
29
+ def context(self) -> flwr.proto.message_pb2.Context: ...
30
+ @property
31
+ def run(self) -> flwr.proto.run_pb2.Run: ...
32
+ @property
33
+ def fab(self) -> flwr.proto.fab_pb2.Fab: ...
34
+ def __init__(self,
35
+ *,
36
+ context: typing.Optional[flwr.proto.message_pb2.Context] = ...,
37
+ run: typing.Optional[flwr.proto.run_pb2.Run] = ...,
38
+ fab: typing.Optional[flwr.proto.fab_pb2.Fab] = ...,
39
+ ) -> None: ...
40
+ def HasField(self, field_name: typing_extensions.Literal["context",b"context","fab",b"fab","run",b"run"]) -> builtins.bool: ...
41
+ def ClearField(self, field_name: typing_extensions.Literal["context",b"context","fab",b"fab","run",b"run"]) -> None: ...
42
+ global___PullSimulationInputsResponse = PullSimulationInputsResponse
43
+
44
+ class PushSimulationOutputsRequest(google.protobuf.message.Message):
45
+ """PushSimulationOutputs messages"""
46
+ DESCRIPTOR: google.protobuf.descriptor.Descriptor
47
+ RUN_ID_FIELD_NUMBER: builtins.int
48
+ CONTEXT_FIELD_NUMBER: builtins.int
49
+ run_id: builtins.int
50
+ @property
51
+ def context(self) -> flwr.proto.message_pb2.Context: ...
52
+ def __init__(self,
53
+ *,
54
+ run_id: builtins.int = ...,
55
+ context: typing.Optional[flwr.proto.message_pb2.Context] = ...,
56
+ ) -> None: ...
57
+ def HasField(self, field_name: typing_extensions.Literal["context",b"context"]) -> builtins.bool: ...
58
+ def ClearField(self, field_name: typing_extensions.Literal["context",b"context","run_id",b"run_id"]) -> None: ...
59
+ global___PushSimulationOutputsRequest = PushSimulationOutputsRequest
60
+
61
+ class PushSimulationOutputsResponse(google.protobuf.message.Message):
62
+ DESCRIPTOR: google.protobuf.descriptor.Descriptor
63
+ def __init__(self,
64
+ ) -> None: ...
65
+ global___PushSimulationOutputsResponse = PushSimulationOutputsResponse