Federated Learning Code In Python For Prime