Fix case sensitivity in AzureVMCluster GPU instance checks (#270)

dask · Mar 23, 2021 · bc949ee · bc949ee
1 parent efbca26
commit bc949ee
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/dask_cloudprovider/azure/azurevm.py b/dask_cloudprovider/azure/azurevm.py
@@ -463,7 +463,9 @@ def __init__(
         )
         if self.scheduler_vm_size is None:
             self.scheduler_vm_size = self.vm_size
-        self.gpu_instance = "NC" in self.vm_size or "ND" in self.vm_size
+        self.gpu_instance = (
+            "_NC" in self.vm_size.upper() or "_ND" in self.vm_size.upper()
+        )
         self.vm_image = self.config.get("vm_image")
         for key in vm_image:
             self.vm_image[key] = vm_image[key]