diff --git a/tests/unit/test_router.py b/tests/unit/test_router.py index 5b0312dda93a1e4266486470e82f0ef5780b7902..b2b87b9339ec6de935a8ef9fa6ffd03f15e5dc42 100644 --- a/tests/unit/test_router.py +++ b/tests/unit/test_router.py @@ -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(), @@ -240,7 +242,9 @@ class TestIndexEncoders: 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)