diff --git a/build.py b/build.py index aaa6b8b965..9b61dd5182 100755 --- a/build.py +++ b/build.py @@ -616,9 +616,7 @@ def pytorch_cmake_args(images): if "pytorch" in images: image = images["pytorch"] else: - image = "nvcr.io/nvidia/pytorch:{}-py3".format( - FLAGS.upstream_container_version - ) + image = "nvcr.io/nvidia/pytorch:{}-py3".format(FLAGS.upstream_container_version) cargs = [ cmake_backend_arg("pytorch", "TRITON_PYTORCH_DOCKER_IMAGE", None, image), ] @@ -651,7 +649,6 @@ def onnxruntime_cmake_args(images, library_paths): ) ) - if target_platform() == "windows": if "base" in images: cargs.append( @@ -767,9 +764,7 @@ def tensorflow_cmake_args(images, library_paths): FLAGS.upstream_container_version ) extra_args = [ - cmake_backend_arg( - backend_name, "TRITON_TENSORFLOW_DOCKER_IMAGE", None, image - ) + cmake_backend_arg(backend_name, "TRITON_TENSORFLOW_DOCKER_IMAGE", None, image) ] return extra_args