Skip to content

Commit c9b3ccf

Browse files
committed
refactor: enhance envs
Signed-off-by: thxCode <[email protected]>
1 parent 6d771e2 commit c9b3ccf

File tree

3 files changed

+9
-6
lines changed

3 files changed

+9
-6
lines changed

gpustack_runner/cmds/images.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1094,7 +1094,7 @@ def list_images(**kwargs) -> list[PlatformedImage]:
10941094
name = img.name
10951095
if not name:
10961096
continue
1097-
if namespace := envs.GPUSTACK_RUNNER_DEFAULT_IMAGE_NAMESPACE:
1097+
if namespace := envs.GPUSTACK_RUNNER_DEFAULT_CONTAINER_NAMESPACE:
10981098
name = name.replace("gpustack/", f"{namespace}/")
10991099
img.name = name
11001100
if name not in image_names_index:

gpustack_runner/envs.py

Lines changed: 7 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -9,21 +9,24 @@
99

1010
# Global
1111

12-
GPUSTACK_RUNNER_DEFAULT_IMAGE_NAMESPACE: str | None = None
12+
GPUSTACK_RUNNER_DEFAULT_CONTAINER_NAMESPACE: str | None = None
1313
"""
14-
Namespace for default runner images.
14+
Namespace for default runner container images.
1515
If not set, it should be "gpustack".
1616
"""
1717

1818
# --8<-- [start:env-vars-definition]
1919

2020
variables: dict[str, Callable[[], Any]] = {
2121
# Global
22-
"GPUSTACK_RUNNER_DEFAULT_IMAGE_NAMESPACE": lambda: trim_str(
22+
"GPUSTACK_RUNNER_DEFAULT_CONTAINER_NAMESPACE": lambda: trim_str(
2323
getenvs(
2424
keys=[
25-
"GPUSTACK_RUNNER_DEFAULT_IMAGE_NAMESPACE",
25+
"GPUSTACK_RUNNER_DEFAULT_CONTAINER_NAMESPACE",
2626
## Compatible with gpustack/gpustack_runtime.
27+
"GPUSTACK_RUNTIME_DEPLOY_DEFAULT_CONTAINER_NAMESPACE",
28+
## Legacy compatibility.
29+
"GPUSTACK_RUNNER_DEFAULT_IMAGE_NAMESPACE",
2730
"GPUSTACK_RUNTIME_DEPLOY_DEFAULT_IMAGE_NAMESPACE",
2831
],
2932
),

gpustack_runner/runner.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -239,7 +239,7 @@ def list_runners(**kwargs) -> Runners | list[dict]:
239239
json_list = json.load(f)
240240
runners = []
241241
for item in json_list:
242-
if namespace := envs.GPUSTACK_RUNNER_DEFAULT_IMAGE_NAMESPACE:
242+
if namespace := envs.GPUSTACK_RUNNER_DEFAULT_CONTAINER_NAMESPACE:
243243
docker_image = item["docker_image"]
244244
docker_image = docker_image.replace("gpustack/", f"{namespace}/")
245245
item["docker_image"] = docker_image

0 commit comments

Comments
 (0)