update

Oneflow-Inc · Sep 4, 2024 · 8cfd032 · 8cfd032
1 parent 942556f
commit 8cfd032
Showing 1 changed file with 2 additions and 6 deletions.
diff --git a/libai/utils/distributed.py b/libai/utils/distributed.py
@@ -76,16 +76,12 @@ def _init_distributed_env(self, cfg):
             try:
                 import oneflow_npu
             except ImportError:
-                raise ImportError(
-                    "'oneflow_npu' is missing. Install it to use NPU devices."
-                )
+                raise ImportError("'oneflow_npu' is missing. Install it to use NPU devices.")
         elif self._device_type == "xpu":
             try:
                 import oneflow_xpu
             except ImportError:
-                raise ImportError(
-                    "'oneflow_xpu' is missing. Install it to use NPU devices."
-                )
+                raise ImportError("'oneflow_xpu' is missing. Install it to use NPU devices.")
         elif self._device_type not in ("cuda", "npu", "xpu", "cpu"):
             raise NotImplementedError(f"Unsupported device {self._device_type}")