Archives/petals
2
0
Fork 0
mirror of https://github.com/bigscience-workshop/petals synced 2024-10-31 09:20:41 +00:00
Code Issues Packages Projects Releases Wiki Activity

Fix running speed test

Browse Source
This commit is contained in:
Aleksandr Borzunov 2022-11-30 04:39:47 +00:00
parent 24a60985c7
commit 3ae859858f
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
src/petals/server/throughput.py
View File

@ -90,7 +90,7 @@ def measure_throughput_info() -> ThroughputInfo:
def measure_network_rps(config: BloomConfig) -> float:
proc = subprocess.run([SPEED_TEST_PATH, "--json"], capture_output=True)
proc = subprocess.run("python -m petals.cli.speed_test --json", shell=True, capture_output=True)
if proc.returncode != 0:
raise RuntimeError(f"Failed to measure network throughput (stdout: {proc.stdout}, stderr: {proc.stderr})")
network_info = json.loads(proc.stdout)