From 922ac24fc5bcfdabb9c0306bfe8d30f89eecbbc0 Mon Sep 17 00:00:00 2001 From: Andrew Date: Thu, 29 Aug 2024 00:03:30 -0700 Subject: [PATCH] lint --- sky/provision/kubernetes/utils.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/sky/provision/kubernetes/utils.py b/sky/provision/kubernetes/utils.py index ec5b1ea4b3b..8ac3ab1d4ca 100644 --- a/sky/provision/kubernetes/utils.py +++ b/sky/provision/kubernetes/utils.py @@ -111,7 +111,8 @@ def get_gke_accelerator_name(accelerator: str) -> str: # H100 is named as H100-80GB in GKE. accelerator = 'H100-80GB' if accelerator in ('A100-80GB', 'L4', 'H100-80GB', 'H100-MEGA-80GB'): - # A100-80GB, L4, H100-80GB and H100-MEGA-80GB have a different name pattern. + # A100-80GB, L4, H100-80GB and H100-MEGA-80GB + # have a different name pattern. return 'nvidia-{}'.format(accelerator.lower()) else: return 'nvidia-tesla-{}'.format(accelerator.lower())