From 3296d87c9632098be2cceed9a7306dfc9270b1cf Mon Sep 17 00:00:00 2001
From: Andrey Velichkevich <andrey.velichkevich@gmail.com>
Date: Fri, 13 Mar 2026 23:42:18 +0000
Subject: [PATCH] fix(ci): Generate valid release version for Python package

Signed-off-by: Andrey Velichkevich <andrey.velichkevich@gmail.com>
---
 VERSION                                         | 2 +-
 api/python_api/kubeflow_trainer_api/__init__.py | 2 +-
 hack/python-api/gen-api.sh                      | 2 +-
 3 files changed, 3 insertions(+), 3 deletions(-)

diff --git a/VERSION b/VERSION
index 852700e118..773e263035 100644
--- a/VERSION
+++ b/VERSION
@@ -1 +1 @@
-v2.1.0
\ No newline at end of file
+v2.2.0-rc.0
\ No newline at end of file
diff --git a/api/python_api/kubeflow_trainer_api/__init__.py b/api/python_api/kubeflow_trainer_api/__init__.py
index 8bd5e8160e..f9f9881e26 100644
--- a/api/python_api/kubeflow_trainer_api/__init__.py
+++ b/api/python_api/kubeflow_trainer_api/__init__.py
@@ -13,4 +13,4 @@
 # limitations under the License.
 
 
-__version__ = "2.1.0"
+__version__ = "2.2.0rc0"
diff --git a/hack/python-api/gen-api.sh b/hack/python-api/gen-api.sh
index 3ceed1064c..bad3ee5028 100755
--- a/hack/python-api/gen-api.sh
+++ b/hack/python-api/gen-api.sh
@@ -33,7 +33,7 @@ if [ ! -f "${VERSION_FILE}" ]; then
   echo "Missing VERSION file at ${VERSION_FILE}"
   exit 1
 fi
-API_VERSION=$(sed 's/^v//' "${VERSION_FILE}")
+API_VERSION=$(sed -e 's/^v//' -e 's/-rc\.\([0-9]*\)/rc\1/' "${VERSION_FILE}")
 API_OUTPUT_PATH="api/python_api"
 PKG_ROOT="${API_OUTPUT_PATH}/kubeflow_trainer_api"