Remove a stale comment, we plan to keep federated_map to make it easier to e.g. support tf.functions directly.
@@ -436,9 +436,6 @@ def build_model_delta_optimizer_process(
client_outputs.weights_delta,
weight=weight_denom)
- # TODO(b/123408447): remove tff.federated_map and call
- # tf_server_update directly once T <-> T@SERVER isomorphism is
- # supported.
server_state = tff.federated_map(
tf_server_update, (server_state, round_model_delta,
new_delta_aggregate_state, new_broadcaster_state))
