Skip to content

Commit 3b82608

Browse files
committed
Rename rt_base to runtime var
1 parent 50cad0b commit 3b82608

File tree

1 file changed

+9
-9
lines changed

1 file changed

+9
-9
lines changed

build.py

Lines changed: 9 additions & 9 deletions
Original file line numberDiff line numberDiff line change
@@ -1227,12 +1227,12 @@ def create_dockerfile_linux(
12271227
repoagents,
12281228
caches,
12291229
endpoints,
1230-
rt_base_image=None,
1230+
runtime_image=None,
12311231
):
12321232
base_image = argmap["BASE_IMAGE"]
12331233
# If runtime base image is provided, use it as the base image
1234-
if rt_base_image:
1235-
base_image = rt_base_image
1234+
if runtime_image:
1235+
base_image = runtime_image
12361236

12371237
df = """
12381238
ARG TRITON_VERSION={}
@@ -1654,12 +1654,12 @@ def change_default_python_version_rhel(version):
16541654

16551655

16561656
def create_dockerfile_windows(
1657-
ddir, dockerfile_name, argmap, backends, repoagents, caches, rt_base_image=None
1657+
ddir, dockerfile_name, argmap, backends, repoagents, caches, runtime_image=None
16581658
):
16591659
base_image = argmap["BASE_IMAGE"]
16601660
# If runtime base image is provided, use it as the base image
1661-
if rt_base_image:
1662-
base_image = rt_base_image
1661+
if runtime_image:
1662+
base_image = runtime_image
16631663
df = """
16641664
ARG TRITON_VERSION={}
16651665
ARG TRITON_CONTAINER_VERSION={}
@@ -1768,7 +1768,7 @@ def create_build_dockerfiles(
17681768
backends,
17691769
repoagents,
17701770
caches,
1771-
images.get("rt_base"),
1771+
images.get("runtime"),
17721772
)
17731773
else:
17741774
create_dockerfile_linux(
@@ -1779,7 +1779,7 @@ def create_build_dockerfiles(
17791779
repoagents,
17801780
caches,
17811781
endpoints,
1782-
images.get("rt_base"),
1782+
images.get("runtime"),
17831783
)
17841784

17851785
# Dockerfile used for the creating the CI base image.
@@ -2964,7 +2964,7 @@ def enable_all():
29642964
"pytorch",
29652965
"tensorflow",
29662966
"tensorflow2",
2967-
"rt_base",
2967+
"runtime",
29682968
],
29692969
"unsupported value for --image",
29702970
)

0 commit comments

Comments
 (0)