Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Fix Aggregator / Assigner leaky abstraction #1301

Open
wants to merge 2 commits into
base: develop
Choose a base branch
from

Conversation

ishaileshpant
Copy link
Collaborator

@ishaileshpant ishaileshpant commented Jan 23, 2025

#FF0000

N.B - This PR has to go after #1226


In PR #1226 task_group is passed to Aggregator and awareness of task_group name is leaked b/w Assigner and Aggregator, to fix this Aggregator should rely on Assigner instead of hardcoding task_group names across multiple classes

@ishaileshpant ishaileshpant changed the title [WIP] Fix Aggregator / Assinger leaky abstraction [WIP] Fix Aggregator / Assigner leaky abstraction Jan 23, 2025
@ishaileshpant ishaileshpant force-pushed the fix_leaky_abstraction branch 2 times, most recently from f40af12 to f2c1606 Compare January 23, 2025 13:20
@ishaileshpant ishaileshpant marked this pull request as ready for review January 23, 2025 13:21
@ishaileshpant ishaileshpant force-pushed the fix_leaky_abstraction branch 3 times, most recently from 662f997 to ea1541b Compare January 23, 2025 17:01
- update all the sub-classes that use task_groups to use the decorator
- update fedeval sample workspace to use default assigner, tasks and aggregator
- use of federated-evaluation/aggregator.yaml for FedEval specific workspace example to use round_number as 1
- removed assigner and tasks yaml from defaults/federated-evaluation, superseded by default assigner/tasks
- Rebase 21-Jan-2025.2
- added additional checks for assigner sub-classes that might not have task_groups
- Addressing review comments
- Updated existing test cases for Assigner sub-classes
- Remove hard-coded setting in assigner for torch_cnn_mnist ws, refer to default as in other Workspaces
- Use aggregator supplied --task_group to override the assinger selected_task_group
- update existing test cases of aggregator cli
- add test cases for the decorator
Signed-off-by: Shailesh Pant <[email protected]>
- Added is_task_group_evaluation function in Assigner class
- Fix existing aggregator interface test cases
- rebased 23-Jan.2
Signed-off-by: Shailesh Pant <[email protected]>
@ishaileshpant ishaileshpant changed the title [WIP] Fix Aggregator / Assigner leaky abstraction Fix Aggregator / Assigner leaky abstraction Jan 23, 2025
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

Successfully merging this pull request may close these issues.

1 participant