diff --git a/studio/backend/core/training/worker.py b/studio/backend/core/training/worker.py index 0d1e51343d..f52dfc48f1 100644 --- a/studio/backend/core/training/worker.py +++ b/studio/backend/core/training/worker.py @@ -749,8 +749,10 @@ def _nvidia_classify_spark_unified_memory(props: Any) -> tuple[str, bool]: if getattr(props, "is_integrated", 0): return "is_integrated", True name_upper = (getattr(props, "name", "") or "").upper() + import re for token in ("GB10", "GB110", "JMJWOA", "N1X", "DGX SPARK"): - if token in name_upper: + # Whole-token match so "GB10" does not match a discrete "GB100"/"GB10X". + if re.search(r"(? timeout = 5, ) _names_u = (_smi.stdout or "").upper() + import re as _re _spark_smi = any( - t in _names_u for t in ("GB10", "GB110", "JMJWOA", "N1X", "DGX SPARK") + _re.search(r"(?