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

for more information, see https://pre-commit.ci
deepmodeling · Jan 9, 2025 · a30053f · a30053f
1 parent b9bdee5
commit a30053f
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/deepmd/pt/utils/dataset.py b/deepmd/pt/utils/dataset.py
@@ -71,7 +71,7 @@ def get_frame_index(self):
                 row_indices_global = np.where(frames_with_elem)[0] + base_offset
                 element_counts[elem]["frames"] += len(row_indices)
                 element_counts[elem]["indices"].extend(row_indices_global.tolist())
-            base_offset += element_data.shape[0] 
+            base_offset += element_data.shape[0]
         element_counts = dict(element_counts)
         return element_counts