Allow to specify release via env

dstackai · Jan 25, 2024 · 8f0fd42 · 8f0fd42
1 parent 3b3309f
commit 8f0fd42
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 1 deletion.
diff --git a/src/dstack/_internal/core/backends/base/compute.py b/src/dstack/_internal/core/backends/base/compute.py
@@ -235,7 +235,7 @@ def get_latest_runner_build() -> Optional[str]:
 
 
 def get_dstack_gateway_wheel(build: str) -> str:
-    channel = "release" if version.__is_release__ else "stgn"
+    channel = "release" if settings.DSTACK_RELEASE else "stgn"
     return f"https://dstack-gateway-downloads.s3.amazonaws.com/{channel}/dstack_gateway-{build}-py3-none-any.whl"
 
 

diff --git a/src/dstack/_internal/settings.py b/src/dstack/_internal/settings.py
@@ -3,3 +3,4 @@
 from dstack import version
 
 DSTACK_VERSION = os.getenv("DSTACK_VERSION", version.__version__)
+DSTACK_RELEASE = os.getenv("DSTACK_RELEASE") is not None or version.__is_release__