check stat.py

deepmodeling · Jan 10, 2025 · 4ce9cfb · 4ce9cfb
1 parent 58647f3
commit 4ce9cfb
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/deepmd/pt/utils/stat.py b/deepmd/pt/utils/stat.py
@@ -161,7 +161,7 @@ def finalize_stats(sys_stat):
 
         # get frame index
         if datasets[0].mixed_type and enable_element_completion:
-            element_counts, type_map = dataset.get_frame_index_for_elements()
+            element_counts, type_name = dataset.get_frame_index_for_elements()
             for new_idx, elem_name in type_name.items():
                 if new_idx not in global_type_name:
                     global_type_name[new_idx] = elem_name