Merge pull request #23274 from ayaka14732:mypy-error

PiperOrigin-RevId: 668010899

build/rocm/run_single_gpu.py

@@ -169,7 +169,7 @@ def run_parallel(all_testmodules, p, c):
 
 
 def find_num_gpus():
-  cmd = ["lspci|grep 'controller\|accel'|grep 'AMD/ATI'|wc -l"]
+  cmd = [r"lspci|grep 'controller\|accel'|grep 'AMD/ATI'|wc -l"]
   _, _, stdout = run_shell_command(cmd, shell=True)
   return int(stdout)
 

build/rocm/tools/build_wheels.py

@@ -101,7 +101,7 @@ def build_jax_wheel(jax_path, python_version):
     env["PATH"] = "%s:%s" % (py_bin, env["PATH"])
 
     LOG.info("Running %r from cwd=%r" % (cmd, jax_path))
-    pattern = re.compile("Successfully built jax-.+ and (jax-.+\.whl)\n")
+    pattern = re.compile(r"Successfully built jax-.+ and (jax-.+\.whl)\n")
 
     _run_scan_for_output(cmd, pattern, env=env, cwd=jax_path, capture="stdout")
 

build/rocm/tools/symbols.py

@@ -42,7 +42,7 @@ def main():
 def highest_for_file(sofile):
     output = subprocess.check_output(["objdump", "-T", sofile])
 
-    r = re.compile("\(GLIBC_(.*)\)")
+    r = re.compile(r"\(GLIBC_(.*)\)")
     versions = {}
 
     for line in output.decode("utf-8").split("\n"):