adap / flower

Flower: A Friendly Federated Learning Framework
https://flower.ai
Apache License 2.0
4.92k stars 848 forks source link

fix(framework) Adjust framework name in templates docstrings #4127

Closed jafermarq closed 2 weeks ago

jafermarq commented 3 weeks ago

With the changes in #3894 we dynamically indicate the framework used to generate each template. However, it uses an all-caps representation. This PR sets the framework string as defined in the MlFramework enum.

Effectively this means the Flower apps will be created showing:

"""my_test_app: A Flower / NumPy app."""

instead of:

"""my_test_app: A Flower / NUMPY app."""