Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
feat: Support for setting aggregation for (Hybrid)RouteLayer (#202)
* Enable to set top_k for RouteLayer * Add top_k unit test for RouteLayer * Added support for setting different aggregation method for HybridRL * Add unit test to verify that aggregation works * Add support for setting aggregation in RouteLayer * Fix bug where agg method not used in RL * Add tests for agg * Linted code * Updated codecov file * Change aggregation method names to lower case --------- Co-authored-by: James Briggs <35938317+jamescalam@users.noreply.github.com>
- Loading branch information