NousResearch/atropos
1
0
Fork 0
mirror of https://github.com/NousResearch/atropos.git synced 2026-04-25 17:10:42 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

clean vllm tonight

Browse source
This commit is contained in:
Jai Suphavadeeprasit 2026-03-23 11:28:40 -07:00
parent 75a032bf3e
commit fae87dcaaa
2 changed files with 1 additions and 6 deletions
Download patch file Download diff file Expand all files Collapse all files

6
example_trainer/api.py
View file

@ -99,11 +99,7 @@ def get_batch(url: str = "http://localhost:8000"):
Raises:
RuntimeError: If trainer is not registered or other API error
"""
response = requests.get(
f"{url}/batch",
timeout=10,
)
data = response.json()
data = requests.get(f"{url}/batch", timeout=10).json()
# Check if there was an error (trainer not registered)
if data.get("status") == "error":