From 4ce9cfba9493f7dc256091814650ac4d38d971da Mon Sep 17 00:00:00 2001 From: SumGuo Date: Fri, 10 Jan 2025 15:49:28 +0800 Subject: [PATCH] check stat.py --- deepmd/pt/utils/stat.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/deepmd/pt/utils/stat.py b/deepmd/pt/utils/stat.py index 7c0a8aa265..7657e84d75 100644 --- 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