diff --git a/studio/install_python_stack.py b/studio/install_python_stack.py index e3b282a667..8196b03ddb 100644 --- a/studio/install_python_stack.py +++ b/studio/install_python_stack.py @@ -72,7 +72,11 @@ def _detect_rocm_version() -> tuple[int, int] | None: if result.returncode == 0: raw = result.stdout.decode().strip().split("\n")[0] parts = raw.split(".") - if len(parts) >= 2 and parts[0].isdigit() and parts[1].split("-")[0].isdigit(): + if ( + len(parts) >= 2 + and parts[0].isdigit() + and parts[1].split("-")[0].isdigit() + ): return int(parts[0]), int(parts[1].split("-")[0]) except Exception: pass