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

for more information, see https://pre-commit.ci
deepmodeling · Oct 16, 2024 · 0b0d943 · 0b0d943
1 parent 6acfd3b
commit 0b0d943
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/deepmd/pt/entrypoints/main.py b/deepmd/pt/entrypoints/main.py
@@ -113,7 +113,9 @@ def get_trainer(
         elif gloo_available:
             backend = "gloo"
         else:
-            raise RuntimeError("No suitable backend found. Neither NCCL nor Gloo is available.")
+            raise RuntimeError(
+                "No suitable backend found. Neither NCCL nor Gloo is available."
+            )
         dist.init_process_group(backend=backend)
 
     def prepare_trainer_input_single(