diff --git a/src/codeweaver/providers/optimize.py b/src/codeweaver/providers/optimize.py index b0d06089..26174d30 100644 --- a/src/codeweaver/providers/optimize.py +++ b/src/codeweaver/providers/optimize.py @@ -51,7 +51,8 @@ def _nvidia_smi_device_ids() -> list[int]: text=True, timeout=2.0, ) - return [int(line.strip()) for line in out.splitlines() if line.strip().isdigit()] + # Using walrus operator avoids redundant .strip() calls on each line + return [int(stripped) for line in out.splitlines() if (stripped := line.strip()).isdigit()] return []