[pre-commit.ci] auto fixes from pre-commit.com hooks
for more information, see https://pre-commit.ci
This commit is contained in:
parent
5c2aefc6d1
commit
5f7d99b4bd
1 changed files with 3 additions and 1 deletions
|
|
@ -47,7 +47,9 @@ def _flex_is_dgx_spark():
|
|||
|
||||
out = subprocess.run(
|
||||
["nvidia-smi", "--query-gpu=name", "--format=csv,noheader"],
|
||||
capture_output = True, text = True, timeout = 5,
|
||||
capture_output = True,
|
||||
text = True,
|
||||
timeout = 5,
|
||||
)
|
||||
names = (out.stdout or "").upper()
|
||||
return any(t in names for t in ("GB10", "JMJWOA", "N1X", "DGX SPARK", "GB110"))
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue