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>
Showing
- coverage.xml 206 additions, 179 deletionscoverage.xml
- semantic_router/hybrid_layer.py 23 additions, 1 deletionsemantic_router/hybrid_layer.py
- semantic_router/layer.py 29 additions, 2 deletionssemantic_router/layer.py
- tests/unit/test_hybrid_layer.py 43 additions, 0 deletionstests/unit/test_hybrid_layer.py
- tests/unit/test_layer.py 43 additions, 1 deletiontests/unit/test_layer.py
Loading
Please register or sign in to comment