changed function name and add comments

triton-inference-server · Jul 20, 2021 · 2b73bef · 2b73bef
1 parent 5012347
commit 2b73bef
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 4 deletions.
diff --git a/build.py b/build.py
@@ -654,7 +654,7 @@ def create_dockerfile_linux(ddir, dockerfile_name, argmap, backends, repoagents,
 '''.format(argmap['TRITON_VERSION'], argmap['TRITON_CONTAINER_VERSION'],
            argmap['BASE_IMAGE'])
 
-    df += dockerfile_add_installation_linux(argmap, backends)
+    df += dockerfile_prepare_container_linux(argmap, backends)
 
     df += '''
 WORKDIR /opt/tritonserver
@@ -692,8 +692,9 @@ def create_dockerfile_linux(ddir, dockerfile_name, argmap, backends, repoagents,
         dfile.write(df)
 
 
-def dockerfile_add_installation_linux(argmap, backends):
-    # Common steps for producing docker images shared by build.py and compose.py.
+def dockerfile_prepare_container_linux(argmap, backends):
+    # Common steps to produce docker images shared by build.py and compose.py.
+    # Sets enviroment variables, installs dependencies and adds entrypoint
     df = '''
 ARG TRITON_VERSION
 ARG TRITON_CONTAINER_VERSION

diff --git a/compose.py b/compose.py
@@ -71,7 +71,7 @@ def start_gpu_dockerfile(ddir, images, argmap, dockerfile_name, backends):
            images["full"], images["min"])
 
     import build
-    df += build.dockerfile_add_installation_linux(argmap, backends)
+    df += build.dockerfile_prepare_container_linux(argmap, backends)
     # Copy over files
     df += '''
 WORKDIR /opt/tritonserver