From 8f0fd4285a1036234173e3fcd5dbbfc89241932d Mon Sep 17 00:00:00 2001 From: Victor Skvortsov Date: Thu, 25 Jan 2024 16:57:42 +0500 Subject: [PATCH] Allow to specify release via env --- src/dstack/_internal/core/backends/base/compute.py | 2 +- src/dstack/_internal/settings.py | 1 + 2 files changed, 2 insertions(+), 1 deletion(-) diff --git a/src/dstack/_internal/core/backends/base/compute.py b/src/dstack/_internal/core/backends/base/compute.py index 81522b488..0465d22cb 100644 --- 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 index 0fcad9b4f..9247e7ba5 100644 --- 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__