Format AMD with space between percent

This commit is contained in:
Nick Mowen 2022-09-24 18:54:15 -06:00
parent 1b35af4c0d
commit 1d7265f453

View File

@ -806,9 +806,9 @@ def get_amd_gpu_stats() -> dict[str, str]:
for hw in usages: for hw in usages:
if "gpu" in hw: if "gpu" in hw:
results["gpu_usage"] = hw.strip().split(" ")[1] results["gpu_usage"] = f"{hw.strip().split(' ')[1].split(' ')[0]} %"
elif "vram" in hw: elif "vram" in hw:
results["memory_usage"] = hw.strip().split(" ")[1] results["memory_usage"] = f"{hw.strip().split(' ')[1].split(' ')[0]} %"
return results return results
@ -831,7 +831,7 @@ def get_nvidia_gpu_stats() -> dict[str, str]:
logger.error(p.stderr) logger.error(p.stderr)
return None return None
else: else:
usages = p.stdout.split("\n")[1].split(",") usages = p.stdout.split("\n")[1].strip()
results: dict[str, str] = { results: dict[str, str] = {
"name": usages[0], "name": usages[0],
"gpu_usage": usages[1], "gpu_usage": usages[1],