From 1c640b25cb711453a4b00438cb60891a9e37a7b8 Mon Sep 17 00:00:00 2001
From: Dan Foreman-Mackey <danfm@google.com>
Date: Wed, 26 Jun 2024 06:38:51 -0400
Subject: [PATCH] Fix pre-commit formatting errors

It looks like #21916 introduced some whitespace issues that are causing
our pre-commits to fail.
---
 build/build.py               | 2 +-
 build/rocm/run_single_gpu.py | 2 +-
 2 files changed, 2 insertions(+), 2 deletions(-)

diff --git a/build/build.py b/build/build.py
index 587c7bc20..88a0c7e83 100755
--- a/build/build.py
+++ b/build/build.py
@@ -683,7 +683,7 @@ def main():
     "--verbose_failures=true",
     *args.bazel_options,
   )
-  
+
   if args.build_gpu_kernel_plugin == "" and not args.build_gpu_pjrt_plugin:
     build_cpu_wheel_command = [
       *command_base,
diff --git a/build/rocm/run_single_gpu.py b/build/rocm/run_single_gpu.py
index bf37a49ee..13dac5ebb 100755
--- a/build/rocm/run_single_gpu.py
+++ b/build/rocm/run_single_gpu.py
@@ -22,7 +22,7 @@ from concurrent.futures import ThreadPoolExecutor
 
 GPU_LOCK = threading.Lock()
 LAST_CODE = 0
-base_dir="./logs"
+base_dir = "./logs"
 
 def extract_filename(path):
   base_name = os.path.basename(path)