@@ -1220,10 +1220,17 @@ def create_dockerfile_cibase(ddir, dockerfile_name, argmap):
1220
1220
1221
1221
1222
1222
def create_dockerfile_linux (
1223
- ddir , dockerfile_name , argmap , backends , repoagents , caches , endpoints , rt_base_image = None
1223
+ ddir ,
1224
+ dockerfile_name ,
1225
+ argmap ,
1226
+ backends ,
1227
+ repoagents ,
1228
+ caches ,
1229
+ endpoints ,
1230
+ rt_base_image = None
1224
1231
):
1225
1232
base_image = argmap ["BASE_IMAGE" ]
1226
- #If runtime base image is provided, use it as the base image
1233
+ # If runtime base image is provided, use it as the base image
1227
1234
if rt_base_image :
1228
1235
base_image = rt_base_image
1229
1236
@@ -1647,13 +1654,18 @@ def change_default_python_version_rhel(version):
1647
1654
1648
1655
1649
1656
def create_dockerfile_windows (
1650
- ddir , dockerfile_name , argmap , backends , repoagents , caches , rt_base_image = None
1657
+ ddir ,
1658
+ dockerfile_name ,
1659
+ argmap ,
1660
+ backends ,
1661
+ repoagents ,
1662
+ caches ,
1663
+ rt_base_image = None
1651
1664
):
1652
1665
base_image = argmap ["BASE_IMAGE" ]
1653
- #If runtime base image is provided, use it as the base image
1666
+ # If runtime base image is provided, use it as the base image
1654
1667
if rt_base_image :
1655
1668
base_image = rt_base_image
1656
-
1657
1669
df = """
1658
1670
ARG TRITON_VERSION={}
1659
1671
ARG TRITON_CONTAINER_VERSION={}
@@ -1754,8 +1766,6 @@ def create_build_dockerfiles(
1754
1766
FLAGS .build_dir , "Dockerfile.buildbase" , dockerfileargmap
1755
1767
)
1756
1768
1757
-
1758
-
1759
1769
if target_platform () == "windows" :
1760
1770
create_dockerfile_windows (
1761
1771
FLAGS .build_dir ,
@@ -1764,7 +1774,7 @@ def create_build_dockerfiles(
1764
1774
backends ,
1765
1775
repoagents ,
1766
1776
caches ,
1767
- images .get ("rt_base" ),
1777
+ images .get ("rt_base" )
1768
1778
)
1769
1779
else :
1770
1780
create_dockerfile_linux (
@@ -2954,7 +2964,14 @@ def enable_all():
2954
2964
)
2955
2965
fail_if (
2956
2966
parts [0 ]
2957
- not in ["base" , "gpu-base" , "pytorch" , "tensorflow" , "tensorflow2" ,"rt_base" ],
2967
+ not in [
2968
+ "base" ,
2969
+ "gpu-base" ,
2970
+ "pytorch" ,
2971
+ "tensorflow" ,
2972
+ "tensorflow2" ,
2973
+ "rt_base"
2974
+ ],
2958
2975
"unsupported value for --image" ,
2959
2976
)
2960
2977
log ('image "{}": "{}"' .format (parts [0 ], parts [1 ]))
0 commit comments