raise ModuleNotFoundError instead

Signed-off-by: Jinzhe Zeng <[email protected]>
deepmodeling · Oct 8, 2023 · bf5d6c3 · bf5d6c3
1 parent 98aecd5
commit bf5d6c3
Showing 1 changed file with 8 additions and 3 deletions.
diff --git a/deepmd/entrypoints/gui.py b/deepmd/entrypoints/gui.py
@@ -14,13 +14,18 @@ def start_dpgui(*, port: int, bind_all: bool, **kwargs):
         to the network on both IPv4 and IPv6 (where available).
     **kwargs
         additional arguments
+
+    Raises
+    ------
+    ModuleNotFoundError
+        The dpgui package is not installed
     """
     try:
         from dpgui import (
             start_dpgui,
         )
-    except ModuleNotFoundError:
-        raise RuntimeError(
+    except ModuleNotFoundError as e:
+        raise ModuleNotFoundError(
             "To use DP-GUI, please install the dpgui package:\npip install dpgui"
-        )
+        ) from e
     start_dpgui(port=port, bind_all=bind_all)