pytorch / opacus / 17777453786

80%

push

github

Add accountant registry (#784)

Summary:
## Types of changes

- [ ] Bug fix (non-breaking change which fixes an issue)
- [x] New feature (non-breaking change which adds functionality)
- [ ] Breaking change (fix or feature that would cause existing functionality to change)
- [ ] Docs change / refactoring / dependency upgrade

## Motivation and Context / Related issue

Allows to register 3rd party accountants to be used in methods like `get_noised_multiplier` and generally introduces an extra extension point.

Closes https://github.com/pytorch/opacus/issues/780

## How Has This Been Tested (if it applies)

Used in our internal implementation with the accountant from [Felipe-Gomez/riskcal](https://github.com/Felipe-Gomez/riskcal).

## Checklist

- [ ] The documentation is up-to-date with the changes I made.
- [x] I have read the **CONTRIBUTING** document and completed the CLA (see **CONTRIBUTING**).
- [x] All tests passed, and additional code has been covered with new tests.

Pull Request resolved: https://github.com/pytorch/opacus/pull/784

Reviewed By: iden-kalemaj

Differential Revision: D81679360

fbshipit-source-id: edda583c7

50 of 53 new or added lines in 3 files covered. (94.34%)

5634 of 7007 relevant lines covered (80.41%)

1.79 hits per line