NousResearch/atropos
1
0
Fork 0
mirror of https://github.com/NousResearch/atropos.git synced 2026-04-19 12:57:58 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

fix error in function inference_node_wandb_watcher.py

Browse source
This commit is contained in:
Micke 2025-06-27 22:13:37 +02:00 • committed by GitHub
parent 4769eeb4a6
commit af57208da2
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
1 changed files with 3 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

6
atroposlib/cli/inference_node_wandb_watcher.py
View file

@ -30,7 +30,7 @@ def run(api_addr, tp, node_num):
break
curr_step = 0
health_statuses = {
f"server/server_heath_{node_num}_{i}": 0.0 for i in range(8 // tp)
f"server/server_health_{node_num}_{i}": 0.0 for i in range(8 // tp)
}
while True:
data = requests.get(f"{api_addr}/status").json()
@ -45,11 +45,11 @@ def run(api_addr, tp, node_num):
health_status = requests.get(
f"http://localhost:{9000 + i}/health_generate"
).status_code
health_statuses[f"server/server_heath_{node_num}_{i}"] = (
health_statuses[f"server/server_health_{node_num}_{i}"] = (
1 if health_status == 200 else 0
)
except (requests.exceptions.ConnectionError, requests.exceptions.Timeout):
health_statuses[f"server/server_heath_{node_num}_{i}"] = 0
health_statuses[f"server/server_health_{node_num}_{i}"] = 0
def main():