Pre-commit fix

triton-inference-server · Dec 13, 2023 · d119601 · d119601
1 parent ef1546f
commit d119601
Showing 1 changed file with 2 additions and 7 deletions.
diff --git 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