fix: set gpu_profile driver to None for non-GPU node pools

diamondpowell · diamondpowell · commit 66093d7fc4ef · 2026-04-20T21:06:25.000-04:00
diff --git a/modules/python/clients/aks_client.py b/modules/python/clients/aks_client.py
@@ -314,7 +314,7 @@ def create_node_pool(
                     "os_disk_type": "Managed",
                     "nodeLabels": {"gpu": "true"} if gpu_node_pool else {},
                     "gpu_profile": {
-                        "driver": "None" if gpu_node_pool and vm_size == "Standard_ND96asr_v4" else "Install",
+                        "driver": "None" if not gpu_node_pool or vm_size == "Standard_ND96asr_v4" else "Install",
                     },
                 }
 

Original file line number	Diff line number	Diff line change
`@@ -314,7 +314,7 @@ def create_node_pool(`
`314`	`314`	`"os_disk_type": "Managed",`
`315`	`315`	`"nodeLabels": {"gpu": "true"} if gpu_node_pool else {},`
`316`	`316`	`"gpu_profile": {`
`317`		`- "driver": "None" if gpu_node_pool and vm_size == "Standard_ND96asr_v4" else "Install",`
	`317`	`+ "driver": "None" if not gpu_node_pool or vm_size == "Standard_ND96asr_v4" else "Install",`
`318`	`318`	`},`
`319`	`319`	`}`
`320`	`320`