Remove incorrect type annotations.
Remove incorrect type annotations.

PiperOrigin-RevId: 392749561
Change-Id: I79a1475c648b674309f48dc61d8015460a2f5d08

google-research/federated,google-research/federated