bug fix

Author: OutisLi

deepmd/utils/data.py

@@ -139,7 +139,12 @@ def __init__(
         # set modifier
         self.modifier = modifier
         # calculate prefix sum for get_item method
-        frames_list = [self._get_nframes(item) for item in self.dirs]
+        frames_list = [
+            self._load_set(item)["coord"].shape[0]
+            if isinstance(item, DPH5Path)
+            else self._get_nframes(item)
+            for item in self.dirs
+        ]
         self.nframes = np.sum(frames_list)
         # The prefix sum stores the range of indices contained in each directory, which is needed by get_item method
         self.prefix_sum = np.cumsum(frames_list).tolist()