diff --git a/atroposlib/tests/test_server_logprobs.py b/atroposlib/tests/test_server_logprobs.py index 2da50b42..8cbd84ad 100644 --- a/atroposlib/tests/test_server_logprobs.py +++ b/atroposlib/tests/test_server_logprobs.py @@ -41,7 +41,9 @@ class _FakeAPIServer(APIServer): class _FakeRoutedServer: - def __init__(self, name: str, train_slots: int, eval_slots: int, healthy: bool = True): + def __init__( + self, name: str, train_slots: int, eval_slots: int, healthy: bool = True + ): self.name = name self.server_healthy = healthy self.sem = AsyncSemWithAdaptiveWeight(4)