fix: fix a typo in deepmodeling#702

Signed-off-by: Jinzhe Zeng <[email protected]>
njzjz · Sep 2, 2024 · 7e8ca4f · 7e8ca4f
1 parent f4d0adf
commit 7e8ca4f
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/dpdata/deepmd/comp.py b/dpdata/deepmd/comp.py
@@ -180,7 +180,7 @@ def dump(folder, data, set_size=5000, comp_prec=np.float32, remove_sets=True):
         dtypes = dpdata.system.LabeledSystem.DTYPES
     else:
         dtypes = dpdata.system.System.DTYPES
-    for dtype in dpdata.system.LabeledSystem.DTYPES:
+    for dtype in dtypes:
         if dtype.name in (
             "atom_numbs",
             "atom_names",