tune: delete AR

.cloud-build/cleanup/resource_cleanup_manager.py

@@ -227,7 +227,19 @@ class ArtifactRegistryCleanupManager(ResourceCleanupManager):
     vertex_ai_resource = "Artifact Registry"
 
     def list(self) -> Any:
-        return ["my-private-repo"]
+        import subprocess
+
+        result = subprocess.run(["gcloud artifacts repositories list --location=us-central1"], 
+                                shell=True, capture_output=True, text=True)
+
+        ret = []
+        lines = result.stdout.split('\n')[2:]
+        for line in lines:
+            repo = line.split(' ')[0]
+            if repo.startswith("my-docker-repo"):
+                ret.append(repo)
+
+        return ret
 
     def delete(self, resource):
         os.system(f"! gcloud artifacts repositories delete {resource} --location=us-central1")

notebooks/official/training/multi_node_ddp_nccl_vertex_training_with_custom_container.ipynb

@@ -1189,6 +1189,8 @@
       "source": [
         "PRIVATE_REPO = \"my-docker-repo\"\n",
         "\n",
+        "! gcloud services enable artifactregistry.googleapis.com\n",
+        "\n",
         "! gcloud artifacts repositories create {PRIVATE_REPO} --repository-format=docker --location={REGION} --description=\"Docker repository\"\n",
         "\n",
         "! gcloud artifacts repositories list"