From 0840c26e94a4a2108514db9eae4aae157e53441a Mon Sep 17 00:00:00 2001
From: "pre-commit-ci[bot]"
 <66853113+pre-commit-ci[bot]@users.noreply.github.com>
Date: Wed, 15 Oct 2025 04:19:21 +0000
Subject: [PATCH] [pre-commit.ci] auto fixes from pre-commit.com hooks

for more information, see https://pre-commit.ci
---
 atroposlib/api/server.py                 |  2 +-
 atroposlib/tests/test_api_compression.py | 11 ++++-------
 2 files changed, 5 insertions(+), 8 deletions(-)

diff --git a/atroposlib/api/server.py b/atroposlib/api/server.py
index 30f948fc..14ee93dc 100644
--- a/atroposlib/api/server.py
+++ b/atroposlib/api/server.py
@@ -82,7 +82,7 @@ class GZipRequestMiddleware:
 
         sent = False
 
-        # needed some odd logic here to handle gzip stream so just returning an empty body 
+        # needed some odd logic here to handle gzip stream so just returning an empty body
         async def new_receive():
             nonlocal sent
             if sent:
diff --git a/atroposlib/tests/test_api_compression.py b/atroposlib/tests/test_api_compression.py
index 3b6da225..25cbdfe0 100644
--- a/atroposlib/tests/test_api_compression.py
+++ b/atroposlib/tests/test_api_compression.py
@@ -105,9 +105,7 @@ class TestAPICompression:
             "scores": [0.5 for _ in range(16)],
             "advantages": [[0.1 for _ in range(512)] for _ in range(16)],
             "ref_logprobs": [[0.2 for _ in range(512)] for _ in range(16)],
-            "messages": [
-                [{"role": "user", "content": "test" * 50}] for _ in range(16)
-            ],
+            "messages": [[{"role": "user", "content": "test" * 50}] for _ in range(16)],
         }
 
         post_response = requests.post(
@@ -157,7 +155,7 @@ class TestAPICompression:
         response = session.get(
             "http://localhost:8000/batch",
             headers={"Accept-Encoding": "gzip"},
-            stream=True
+            stream=True,
         )
 
         assert response.status_code == 200
@@ -201,7 +199,7 @@ class TestAPICompression:
         data = response.json()
 
         uncompressed_json = json.dumps(data)
-        uncompressed_size = len(uncompressed_json.encode('utf-8'))
+        uncompressed_size = len(uncompressed_json.encode("utf-8"))
 
         assert data["batch"] is not None
         batch = data["batch"][0]
@@ -257,8 +255,7 @@ class TestAPICompression:
         assert post_response.status_code == 200
 
         status_response = requests.get(
-            "http://localhost:8000/status-env",
-            json={"env_id": env_id}
+            "http://localhost:8000/status-env", json={"env_id": env_id}
         )
         assert status_response.status_code == 200
         status_data = status_response.json()