Skip to content

Commit

Permalink
chore: lint
Browse files Browse the repository at this point in the history
  • Loading branch information
jamescalam committed Jan 3, 2025
1 parent 0f5ffd1 commit 84d9a7d
Showing 1 changed file with 6 additions and 2 deletions.
8 changes: 6 additions & 2 deletions tests/unit/test_router.py
Original file line number Diff line number Diff line change
Expand Up @@ -218,7 +218,9 @@ def get_test_routers():
],
)
class TestIndexEncoders:
def test_initialization(self, routes, openai_encoder, index_cls, encoder_cls, router_cls):
def test_initialization(
self, routes, openai_encoder, index_cls, encoder_cls, router_cls
):
index = init_index(index_cls)
route_layer = router_cls(
encoder=encoder_cls(),
Expand All @@ -240,7 +242,9 @@ def test_initialization(self, routes, openai_encoder, index_cls, encoder_cls, ro
else 0 == 2
)

def test_initialization_different_encoders(self, encoder_cls, index_cls, router_cls):
def test_initialization_different_encoders(
self, encoder_cls, index_cls, router_cls
):
index = init_index(index_cls)
encoder = encoder_cls()
route_layer = router_cls(encoder=encoder, index=index)
Expand Down

0 comments on commit 84d9a7d

Please sign in to comment.