Skip to content

Feature Election contribution#3876

Merged
chesterxgchen merged 236 commits intoNVIDIA:mainfrom
christofilojohn:main
Mar 17, 2026
Merged

Feature Election contribution#3876
chesterxgchen merged 236 commits intoNVIDIA:mainfrom
christofilojohn:main

Conversation

@christofilojohn
Copy link
Copy Markdown
Contributor

Fixes
Implemented all the changes you requested on file format, tests etc. black-check, isort-check, flake8 pass.

Types of changes

  • Non-breaking change (fix or new feature that would not break existing functionality).
  • Breaking change (fix or new feature that would cause existing functionality to change).
  • New tests added to cover the changes.
  • Quick tests passed locally by running ./runtest.sh.
  • In-line docstrings updated.
  • Documentation updated.

christofilojohn and others added 30 commits November 2, 2025 13:10
Co-authored-by: greptile-apps[bot] <165735046+greptile-apps[bot]@users.noreply.github.com>
Co-authored-by: greptile-apps[bot] <165735046+greptile-apps[bot]@users.noreply.github.com>
Co-authored-by: greptile-apps[bot] <165735046+greptile-apps[bot]@users.noreply.github.com>
Removed leftover testing code

Co-authored-by: greptile-apps[bot] <165735046+greptile-apps[bot]@users.noreply.github.com>
Co-authored-by: greptile-apps[bot] <165735046+greptile-apps[bot]@users.noreply.github.com>
Co-authored-by: greptile-apps[bot] <165735046+greptile-apps[bot]@users.noreply.github.com>
Co-authored-by: greptile-apps[bot] <165735046+greptile-apps[bot]@users.noreply.github.com>
Co-authored-by: greptile-apps[bot] <165735046+greptile-apps[bot]@users.noreply.github.com>
Co-authored-by: greptile-apps[bot] <165735046+greptile-apps[bot]@users.noreply.github.com>
Co-authored-by: greptile-apps[bot] <165735046+greptile-apps[bot]@users.noreply.github.com>
…ng aggregation works correctly in production deployments
…after mask application, and document client_stats exclusion from persistence
…s except block so a workflow exception puts the FL job into a defined abort state rather than silently swallowing the error. - controller.py: wrap stop_controller's JSON file write in try/except OSError so a filesystem failure during teardown is logged but does not propagate and interrupt controller shutdown. - feature_election.py: add empty-client guards (ValueError) to the random and sequential-fallback branches of prepare_data_splits, matching the existing guards in the stratified and dirichlet paths. - feature_election.py: validate that every DataFrame client in simulate_election has the same column labels as the first client; raise ValueError on mismatch instead of silently misaligning feature name reporting.
…tness issues

- executor: consistent scaler in tuning eval; ValueError on unknown fs_method; expand PyImpetus fallback warning
- controller: skip all-zero client masks in _extract_client_data; plateau early-exit in tuning loop
- feature_election: use RandomState (not np.random.seed) in random/non_iid splits; validate params on load_results; plateau early-exit in simulate_election tuning loop
-prepare_data.py uses np.random.seed() (global state) in all three split functions
-feature_election.py __init__ doesn't validate tuning_rounds >= 0
-feature_election.py "dirichlet" alias silently converts with no deprecation warning
-executor.py _handle_tuning_eval gives no log when tuning_mask is absent from shareable
-test.py doesn't validate executor client config args, and has no tests for unknown fs_method, load_results round-trip, or quick_election bad kwargs
@chesterxgchen
Copy link
Copy Markdown
Collaborator

/build

Copy link
Copy Markdown
Collaborator

@chesterxgchen chesterxgchen left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

LGTM

@chesterxgchen chesterxgchen merged commit 10ab636 into NVIDIA:main Mar 17, 2026
28 checks passed
@chesterxgchen
Copy link
Copy Markdown
Collaborator

@christofilojohn the PR merged, thank you for your contribution and patient to go through the PR review process.

@christofilojohn
Copy link
Copy Markdown
Contributor Author

@chesterxgchen Thank you for the support !

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.

2 participants