Skip to content

Commit

Permalink
chore: missed file lint
Browse files Browse the repository at this point in the history
  • Loading branch information
Anush008 committed Mar 16, 2024
1 parent 838767a commit ea73eea
Showing 1 changed file with 3 additions and 1 deletion.
4 changes: 3 additions & 1 deletion tests/unit/test_layer.py
Original file line number Diff line number Diff line change
Expand Up @@ -133,7 +133,9 @@ def get_test_indexes():
@pytest.mark.parametrize("index_cls", get_test_indexes())
class TestRouteLayer:
def test_initialization(self, openai_encoder, routes, index_cls):
route_layer = RouteLayer(encoder=openai_encoder, routes=routes, top_k=10, index=index_cls())
route_layer = RouteLayer(
encoder=openai_encoder, routes=routes, top_k=10, index=index_cls()
)
assert openai_encoder.score_threshold == 0.82
assert route_layer.score_threshold == 0.82
assert route_layer.top_k == 10
Expand Down

0 comments on commit ea73eea

Please sign in to comment.