Update federated_research/targeted_attack test to use `model_fn`.
Update federated_research/targeted_attack test to use `model_fn`.

PiperOrigin-RevId: 428612292
Change-Id: I5cd7c4d10a033357fea276462caeaf093e919658

google-research/federated,google-research/federated