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)