[pre-commit.ci] auto fixes from pre-commit.com hooks

for more information, see https://pre-commit.ci

Author: pre-commit-ci[bot]

build_tools/jax.py

@@ -102,9 +102,9 @@ def setup_jax_extension(
         cxx_flags.append("-g0")
 
     if bool(int(os.getenv("NVTE_UB_WITH_MPI", "0"))):
-        assert os.getenv("MPI_HOME") is not None, (
-            "MPI_HOME=/path/to/mpi must be set when compiling with NVTE_UB_WITH_MPI=1!"
-        )
+        assert (
+            os.getenv("MPI_HOME") is not None
+        ), "MPI_HOME=/path/to/mpi must be set when compiling with NVTE_UB_WITH_MPI=1!"
         mpi_path = Path(os.getenv("MPI_HOME"))
         include_dirs.append(mpi_path / "include")
         cxx_flags.append("-DNVTE_UB_WITH_MPI")