diff --git a/studio/install_python_stack.py b/studio/install_python_stack.py index e4c4576fbf..bbf71b4496 100644 --- a/studio/install_python_stack.py +++ b/studio/install_python_stack.py @@ -112,7 +112,10 @@ def _has_rocm_gpu() -> bool: # rocminfo: look for "Name: gfxNNNN" indicating an actual GPU agent (["rocminfo"], lambda out: "gfx" in out.lower()), # amd-smi list: require "GPU: " data rows, not just a header - (["amd-smi", "list"], lambda out: bool(re.search(r"(?im)^gpu\s*[:\[]\s*\d", out))), + ( + ["amd-smi", "list"], + lambda out: bool(re.search(r"(?im)^gpu\s*[:\[]\s*\d", out)), + ), ): exe = shutil.which(cmd[0]) if not exe: