I got error while compiling the code: I don't know where I am getting issues.
model_fn: Union[Callable[[], tff.learning.Model], tff.learning.models.FunctionalModel]
^
SyntaxError: invalid syntax
from keras import optimizers
from keras.api._v2.keras.optimizers import Adam
from tensorflow_federated.python.learning.algorithms.fed_sgd import DEFAULT_SERVER_OPTIMIZER_FN
from tensorflow_federated.python.learning.models.functional import FunctionalModel
from collections.abc import Callable
from typing import Union
# Define federated learning averaging process
# federated_averaging_process = tff.learning.build_federated_averaging_process(model_fn,
federated_averaging_process = tff.learning.algorithms.build_weighted_fed_avg(
model_fn: Union[Callable[[], tff.learning.Model], tff.learning.models.FunctionalModel]
client_optimizer_fn=Union[tff.learning.optimizers.SGD, Callable[[], tf,keras.optimzers.SGD(learning_rate=0.01]],
#lambda: tf.keras.optimizers.SGD(learning_rate=0.01),
server_optimizer_fn= Union[tff.learning.optimizers.Adam, Callable[[], tf.keras,optimizers.Adam(learning_rate=0.01)]]==DEFAULT_SERVER_OPTIMIZER_FN
#lambda: tf.keras.optimizers.Adam(learning_rate=1.0)
