Update vm image.

mark · mark · commit d9184774267c · 2025-06-14T14:21:10.000-07:00
diff --git a/axlearn/cloud/gcp/vm.py b/axlearn/cloud/gcp/vm.py
@@ -376,7 +376,7 @@ def list_disk_images(creds: Credentials) -> list[str]:
     )
     image_names = []
     for el in images["items"]:
-        if "ubuntu-2004" in el["name"] and "arm" not in el["name"]:
+        if "ubuntu-2204" in el["name"] and "arm" not in el["name"]:
             image_names.append(el["name"])
     return [f"projects/{image_project}/global/images/{name}" for name in image_names]
 

Original file line number	Diff line number	Diff line change
`@@ -376,7 +376,7 @@ def list_disk_images(creds: Credentials) -> list[str]:`
`376`	`376`	`)`
`377`	`377`	`image_names = []`
`378`	`378`	`for el in images["items"]:`
`379`		`- if "ubuntu-2004" in el["name"] and "arm" not in el["name"]:`
	`379`	`+ if "ubuntu-2204" in el["name"] and "arm" not in el["name"]:`
`380`	`380`	`image_names.append(el["name"])`
`381`	`381`	`return [f"projects/{image_project}/global/images/{name}" for name in image_names]`
`382`	`382`