Skip to content

Commit b655f0d

Browse files
committed
Syntax fixes to pass pre-commit check
1 parent d6e151b commit b655f0d

File tree

1 file changed

+26
-9
lines changed

1 file changed

+26
-9
lines changed

build.py

Lines changed: 26 additions & 9 deletions
Original file line numberDiff line numberDiff line change
@@ -1220,10 +1220,17 @@ def create_dockerfile_cibase(ddir, dockerfile_name, argmap):
12201220

12211221

12221222
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
12241231
):
12251232
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
12271234
if rt_base_image:
12281235
base_image = rt_base_image
12291236

@@ -1647,13 +1654,18 @@ def change_default_python_version_rhel(version):
16471654

16481655

16491656
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
16511664
):
16521665
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
16541667
if rt_base_image:
16551668
base_image = rt_base_image
1656-
16571669
df = """
16581670
ARG TRITON_VERSION={}
16591671
ARG TRITON_CONTAINER_VERSION={}
@@ -1754,8 +1766,6 @@ def create_build_dockerfiles(
17541766
FLAGS.build_dir, "Dockerfile.buildbase", dockerfileargmap
17551767
)
17561768

1757-
1758-
17591769
if target_platform() == "windows":
17601770
create_dockerfile_windows(
17611771
FLAGS.build_dir,
@@ -1764,7 +1774,7 @@ def create_build_dockerfiles(
17641774
backends,
17651775
repoagents,
17661776
caches,
1767-
images.get("rt_base"),
1777+
images.get("rt_base")
17681778
)
17691779
else:
17701780
create_dockerfile_linux(
@@ -2954,7 +2964,14 @@ def enable_all():
29542964
)
29552965
fail_if(
29562966
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+
],
29582975
"unsupported value for --image",
29592976
)
29602977
log('image "{}": "{}"'.format(parts[0], parts[1]))

0 commit comments

Comments
 (0)