diff --git a/deepmd/utils/data.py b/deepmd/utils/data.py index ecf4aaeaba..99337aa41d 100644 --- a/deepmd/utils/data.py +++ b/deepmd/utils/data.py @@ -508,7 +508,7 @@ def _make_idx_map(self, atom_type): def _load_type_map(self, sys_path: DPPath) : fname = sys_path / 'type_map.raw' if fname.is_file() : - return fname.load_txt(dtype=str).tolist() + return fname.load_txt(dtype=str, ndmin=1).tolist() else : return None