Skip to content

Commit 5a34348

Browse files
committed
update the return output for easier parsing
1 parent 036d40b commit 5a34348

1 file changed

Lines changed: 2 additions & 2 deletions

File tree

src/utils/core/gpu_install.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -8,7 +8,7 @@
88
from src.utils.log.logger import Logger
99

1010

11-
def fetch_gpu(log: Logger) -> str | None:
11+
def fetch_gpu(log: Logger) -> tuple[str, str | None] | None:
1212
"""Fetch the GPU of the system.
1313
1414
Args:
@@ -31,4 +31,4 @@ def fetch_gpu(log: Logger) -> str | None:
3131
log.logger("e", f"{Err}. Command lspci failed to execute.")
3232
return None
3333
else:
34-
return gpu_name
34+
return gpu_name.decode("utf-8").strip().split("\n")

0 commit comments

Comments
 (0)