From 1b6eecfae12fd25afc0233c266994bc5aaf87c7a Mon Sep 17 00:00:00 2001
From: Jvst Me <git@jvst.me>
Date: Thu, 19 Jun 2025 20:11:31 +0200
Subject: [PATCH] Fix building `dstack` package

---
 .github/workflows/release.yml | 4 +++-
 src/dstack/version.py         | 5 +++++
 2 files changed, 8 insertions(+), 1 deletion(-)

diff --git a/.github/workflows/release.yml b/.github/workflows/release.yml
index 656d0eb8f7..391761e596 100644
--- a/.github/workflows/release.yml
+++ b/.github/workflows/release.yml
@@ -230,10 +230,12 @@ jobs:
       - name: Upload pip package
         run: |
           VERSION=${GITHUB_REF#refs/tags/}
-          BASE_IMAGE=$(cat src/dstack/version.py | grep base_image)
+          BASE_IMAGE=$(cat src/dstack/version.py | grep "base_image = ")
+          BASE_IMAGE_UBUNTU_VERSION=$(cat src/dstack/version.py | grep "base_image_ubuntu_version = ")
           echo "__version__ = \"$VERSION\"" > src/dstack/version.py
           echo "__is_release__ = True" >> src/dstack/version.py
           echo $BASE_IMAGE >> src/dstack/version.py
+          echo $BASE_IMAGE_UBUNTU_VERSION >> src/dstack/version.py
           cp README.md src
           uv build
           uv publish --username ${{ secrets.PYPI_USERNAME }} --password ${{ secrets.PYPI_PASSWORD }}   
diff --git a/src/dstack/version.py b/src/dstack/version.py
index d4255a2301..cc3efc230a 100644
--- a/src/dstack/version.py
+++ b/src/dstack/version.py
@@ -1,3 +1,8 @@
+# WARNING:
+# This file is overwritten when building the dstack package in CI.
+# If you are adding, removing, or renaming variables,
+# remember to update and test the release.yml workflow.
+
 __version__ = "0.0.0"
 __is_release__ = False
 base_image = "0.10"