fix ut

deepmodeling · Dec 2, 2024 · 89a3882 · 89a3882
1 parent fd59449
commit 89a3882
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/deepmd/utils/data.py b/deepmd/utils/data.py
@@ -60,6 +60,8 @@ def __init__(
     ) -> None:
         """Constructor."""
         root = DPPath(sys_path)
+        if not root.is_dir():
+            raise FileNotFoundError(f"System {sys_path} is not found!")
         self.dirs = root.glob(set_prefix + ".*")
         if not len(self.dirs):
             raise FileNotFoundError(f"No {set_prefix}.* is found in {sys_path}")

diff --git a/deepmd/utils/path.py b/deepmd/utils/path.py
@@ -168,8 +168,6 @@ def __init__(self, path: Union[str, Path], mode: str = "r") -> None:
         super().__init__()
         self.mode = mode
         self.path = Path(path)
-        if not self.path.exists():
-            raise FileNotFoundError(f"{self.path} not found")
 
     def load_numpy(self) -> np.ndarray:
         """Load NumPy array.