@@ -1227,12 +1227,12 @@ def create_dockerfile_linux(
1227
1227
repoagents ,
1228
1228
caches ,
1229
1229
endpoints ,
1230
- rt_base_image = None ,
1230
+ runtime_image = None ,
1231
1231
):
1232
1232
base_image = argmap ["BASE_IMAGE" ]
1233
1233
# 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
1236
1236
1237
1237
df = """
1238
1238
ARG TRITON_VERSION={}
@@ -1654,12 +1654,12 @@ def change_default_python_version_rhel(version):
1654
1654
1655
1655
1656
1656
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
1658
1658
):
1659
1659
base_image = argmap ["BASE_IMAGE" ]
1660
1660
# 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
1663
1663
df = """
1664
1664
ARG TRITON_VERSION={}
1665
1665
ARG TRITON_CONTAINER_VERSION={}
@@ -1768,7 +1768,7 @@ def create_build_dockerfiles(
1768
1768
backends ,
1769
1769
repoagents ,
1770
1770
caches ,
1771
- images .get ("rt_base " ),
1771
+ images .get ("runtime " ),
1772
1772
)
1773
1773
else :
1774
1774
create_dockerfile_linux (
@@ -1779,7 +1779,7 @@ def create_build_dockerfiles(
1779
1779
repoagents ,
1780
1780
caches ,
1781
1781
endpoints ,
1782
- images .get ("rt_base " ),
1782
+ images .get ("runtime " ),
1783
1783
)
1784
1784
1785
1785
# Dockerfile used for the creating the CI base image.
@@ -2964,7 +2964,7 @@ def enable_all():
2964
2964
"pytorch" ,
2965
2965
"tensorflow" ,
2966
2966
"tensorflow2" ,
2967
- "rt_base " ,
2967
+ "runtime " ,
2968
2968
],
2969
2969
"unsupported value for --image" ,
2970
2970
)
0 commit comments