Federated Learning Code In Python