Error of atom type mapping when tesing

[J-You]

Hi all,

I created 5 systems for a train, including: Fe, Sn, FeSn, FeSn2, and FeSn5.
I map atom Fe to type-0, Sn to type-1, referring to: https://figshare.com/articles/dataset/Dataset_for_methane_combustion/12973055

All of my type_map.raw follows:
Fe Sn.
I freeze FeSn.pb file successfully.

When I test my model using dp test -m FeSn.pb -s ../data/FeSn -n 20000 --shuffle-test -d TEST, it throws an error:

---------------output of dp test--------------- 
# testing system : ../data/FeSn
Traceback (most recent call last):
  File "/home/why/deepmd-kit/bin/dp", line 10, in <module>
    sys.exit(main())
  File "/home/why/deepmd-kit/lib/python3.9/site-packages/deepmd/main.py", line 79, in main
    test(args)
  File "/home/why/deepmd-kit/lib/python3.9/site-packages/deepmd/test.py", line 43, in test
    err, siz = test_ener(dp, args, append_detail = (cc!=0))
  File "/home/why/deepmd-kit/lib/python3.9/site-packages/deepmd/test.py", line 110, in test_ener
    data = DeepmdData(args.system, args.set_prefix, shuffle_test = args.shuffle_test, type_map = dp.get_type_map())
  File "/home/why/deepmd-kit/lib/python3.9/site-packages/deepmd/Data.py", line 30, in __init__
    atom_type_ = [type_map.index(self.type_map[ii]) for ii in self.atom_type]
  File "/home/why/deepmd-kit/lib/python3.9/site-packages/deepmd/Data.py", line 30, in <listcomp>
    atom_type_ = [type_map.index(self.type_map[ii]) for ii in self.atom_type]
ValueError: 'Sn' is not in list

[njzjz]

See #268.

[J-You]

Thanks for your timely response.
I have noticed #268 before my training.
my json file:

"model": {
        "descriptor" :{
            "type_map":         ["Fe", "Sn"],

[njzjz]

In fact the correct one is

deepmd-kit/examples/water/train/water_se_a.json

Lines 3 to 5 in 59d780c

	"model": {
	"type_map": ["O", "H"],
	"descriptor" :{

[J-You]

sorry for my carelessness