From 534ba3bfa25755eff85b3fbaa43d3ee86d507f89 Mon Sep 17 00:00:00 2001 From: "pre-commit-ci[bot]" <66853113+pre-commit-ci[bot]@users.noreply.github.com> Date: Mon, 27 Jul 2026 23:19:34 +0000 Subject: [PATCH] [pre-commit.ci] auto fixes from pre-commit.com hooks for more information, see https://pre-commit.ci --- studio/install_python_stack.py | 29 ++++++++++++++++------------- 1 file changed, 16 insertions(+), 13 deletions(-) diff --git a/studio/install_python_stack.py b/studio/install_python_stack.py index d27fff0d82..65530a3f00 100644 --- a/studio/install_python_stack.py +++ b/studio/install_python_stack.py @@ -1175,6 +1175,7 @@ def _has_usable_nvidia_gpu() -> bool: cvd = os.environ.get("CUDA_VISIBLE_DEVICES") if cvd is not None and cvd.strip() in ("", "-1"): return False + def _lists_a_gpu(exe: str) -> bool: try: result = subprocess.run( @@ -1200,19 +1201,21 @@ def _has_usable_nvidia_gpu() -> bool: if _path_exe: candidates.append(_path_exe) if IS_WINDOWS: - candidates.extend(( - os.path.join( - os.environ.get("ProgramFiles", r"C:\Program Files"), - "NVIDIA Corporation", - "NVSMI", - "nvidia-smi.exe", - ), - os.path.join( - os.environ.get("SystemRoot", r"C:\Windows"), - "System32", - "nvidia-smi.exe", - ), - )) + candidates.extend( + ( + os.path.join( + os.environ.get("ProgramFiles", r"C:\Program Files"), + "NVIDIA Corporation", + "NVSMI", + "nvidia-smi.exe", + ), + os.path.join( + os.environ.get("SystemRoot", r"C:\Windows"), + "System32", + "nvidia-smi.exe", + ), + ) + ) for _candidate in candidates: if _candidate != _path_exe and not os.path.isfile(_candidate): continue