From c6ba0e98426a535983484d1aa93b5779b6b238f8 Mon Sep 17 00:00:00 2001
From: Pei Zhang <piz@google.com>
Date: Tue, 13 Aug 2024 22:37:01 -0700
Subject: [PATCH] update wheel name (#7722)

---
 infra/ansible/roles/build_srcs/tasks/main.yaml | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/infra/ansible/roles/build_srcs/tasks/main.yaml b/infra/ansible/roles/build_srcs/tasks/main.yaml
index 7592cf2ec6a..544b6b5e158 100644
--- a/infra/ansible/roles/build_srcs/tasks/main.yaml
+++ b/infra/ansible/roles/build_srcs/tasks/main.yaml
@@ -75,11 +75,11 @@
   ansible.builtin.shell: |
     pushd /tmp/staging-wheels
     cp {{ item.dir }}/*.whl .
-    rename -v "s/^{{ item.prefix }}-(.*?)-cp/{{ item.prefix }}-nightly-cp/" *.whl
+    rename -v "s/^{{ item.prefix }}-(.*?)-cp/{{ item.prefix }}-\1+nightly-cp/" *.whl
     mv /tmp/staging-wheels/* /dist/
     popd
 
-    rename -v "s/^{{ item.prefix }}-(.*?)-cp/{{ item.prefix }}-nightly+$(date -u +%Y%m%d)-cp/" *.whl
+    rename -v "s/^{{ item.prefix }}-(.*?)-cp/{{ item.prefix }}-\1+nightly$(date -u +%Y%m%d)-cp/" *.whl
   args:
     executable: /bin/bash
     chdir: "{{ item.dir }}"