update to 2-5 for v6e test name (#425)

GoogleCloudPlatform · Oct 1, 2024 · 7f05c24 · 7f05c24
1 parent e7c5dd4
commit 7f05c24
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/dags/pytorch_xla/r2_5.py b/dags/pytorch_xla/r2_5.py
@@ -203,7 +203,7 @@ def llama():
 
   ci_trillium_4 = task.run_queued_resource_test(
       test_config.JSonnetTpuVmTest.from_pytorch(
-          "pt-nightly-ci-func-v6e-4-1vm",
+          "pt-2-5-ci-func-v6e-4-1vm",
           network=V5_NETWORKS,
           subnetwork=V6E_SUBNETWORKS,
       ),