From 4be3589dda73c9dd0e71b1f9af01bbc48f859234 Mon Sep 17 00:00:00 2001 From: Andrew Aikawa Date: Thu, 2 May 2024 18:32:09 +0000 Subject: [PATCH] pattern match against A100' --- sky/provision/kubernetes/utils.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/sky/provision/kubernetes/utils.py b/sky/provision/kubernetes/utils.py index bc083fc0b17..be67ef2b929 100644 --- a/sky/provision/kubernetes/utils.py +++ b/sky/provision/kubernetes/utils.py @@ -177,7 +177,7 @@ def get_accelerator_from_label_value(cls, value: str) -> str: for canonical_name in canonical_gpu_names: # A100-80G accelerator can be labeled as A100-SXM-80GB or A100-PCIE-80GB - if canonical_name == 'A100-80G' and if re.match(r'A100.*-80G', value): + if canonical_name == 'A100-80G' and re.match(r'A100.*-80G', value): return canonical_name elif canonical_name in value: return canonical_name