More fixes + run on PRs

.github/workflows/lint-and-test.yml

@@ -5,6 +5,7 @@ on:
     branches:
       - main
   workflow_dispatch:
+  pull_request:
 
 jobs:
   build:
@@ -29,7 +30,7 @@ jobs:
         flake8 . --max-line-length=127
     - name: Check formatting with Black
       run: |
-        black --check .
+        black --check --diff .
     - name: Static type checking
       run: |
         mypy .

tests/dummy_server.py

@@ -10,6 +10,7 @@
    tokens in case of an LLM API - and the number of tokens used would only be known after the
    request is processed.
 """
+
 import asyncio
 import json
 import random

tests/test_calls.py

@@ -1,4 +1,5 @@
 """Intended to test the calls module - not performing actual calls to any API"""
+
 from rate_limited.calls import Call
 
 

tests/test_runner.py

@@ -273,11 +273,9 @@ def random_client(url: str, how_many=2, failure_proba: float = 0.2) -> dict:
 
     def validate_num_x_2(result: dict) -> bool:
         return result["output"].count("x") == 2
-
 
     def validate_starts_with_x(result: dict) -> bool:
         return result["output"].startswith("x")
-
 
     validation = validate_num_x_2
     if add_extra_validator:
@@ -298,11 +296,7 @@ def validate_starts_with_x(result: dict) -> bool:
     outputs = [result["output"] for result in results]
     assert outputs == ["xx"] * num_requests
 
-    exceptions_flat = [
-        e
-        for sublist in exceptions
-        for e in sublist
-    ]
+    exceptions_flat = [e for sublist in exceptions for e in sublist]
     assert any(isinstance(e, ValidationError) for e in exceptions_flat)