diff --git a/src/pbfetch/parse/gpu.py b/src/pbfetch/parse/gpu.py index 6b75ecc..019fe3d 100644 --- a/src/pbfetch/parse/gpu.py +++ b/src/pbfetch/parse/gpu.py @@ -1,4 +1,5 @@ from subprocess import Popen, PIPE +from re import sub # import nvidia_smi @@ -30,7 +31,7 @@ def parse_gpu(): if gpu_name: gpu_name = gpu_name.split(": ")[1] gpu_name = gpu_name.strip() - return gpu_name + return sub(r"\(.+\)$", "", gpu_name.strip()) except Exception: pass @@ -45,7 +46,7 @@ def parse_gpu(): gpu_name = gpu_name.replace("OpenGL compatibility profile renderer: ", "") # print(gpu_name) - return gpu_name + return sub(r"\(.+\)$", "", gpu_name.strip()) except Exception: pass @@ -63,7 +64,7 @@ def parse_gpu(): gpu_name = gpu_name[2 : len(gpu_name) - 3] gpu_name = gpu_name.split("VGA compatible controller:")[1].strip() - return gpu_name + return sub(r"\(.+\)$", "", gpu_name.strip()) except Exception as e: print(f"GPU Parse Error: {e}")