Error reported when distillation is performed using the se_2_a descriptor

[JiangXiaoMingSan]

We refer to the example of zbl on GitHub(water/zbl) and use the se_2_a descriptor for distillation to report an error：TypeError: DescrptSeA.compute_input_stats() takes 7 positional arguments but 9 were given. Whether I use deepmdkit-b4 or deepmdkit-b3, this error will occur. The error website :https://bohrium.dp.tech/jobs/detail/15319264, The GitHub URL for submitting the script is https://github.com/JiangXiaoMingSan/se_e2_a-error.git. Thank you for your attention to this issue.

[wanghan-iapcm]

we cannot visit your job on bohirum. Please paste the inputs and the steps to reproduce the issue.

[JiangXiaoMingSan]

Thank you for your reminder. The specific information of the error is as follows. The input file can refer to GitHub: https://github.com/JiangXiaoMingSan/se_e2_a-error.git
Traceback (most recent call last):
File "/opt/deepmd-kit-3.0.0b4/bin/dp", line 10, in
sys.exit(main())
File "/opt/deepmd-kit-3.0.0b4/lib/python3.10/site-packages/deepmd/main.py", line 929, in main
deepmd_main(args)
File "/opt/deepmd-kit-3.0.0b4/lib/python3.10/site-packages/deepmd/tf/entrypoints/main.py", line 72, in main
train_dp(**dict_args)
File "/opt/deepmd-kit-3.0.0b4/lib/python3.10/site-packages/deepmd/tf/entrypoints/train.py", line 186, in train
_do_work(jdata, run_opt, is_compress)
File "/opt/deepmd-kit-3.0.0b4/lib/python3.10/site-packages/deepmd/tf/entrypoints/train.py", line 261, in _do_work
model.build(train_data, stop_batch, origin_type_map=origin_type_map)
File "/opt/deepmd-kit-3.0.0b4/lib/python3.10/site-packages/deepmd/tf/train/trainer.py", line 217, in build
self.model.data_stat(data)
File "/opt/deepmd-kit-3.0.0b4/lib/python3.10/site-packages/deepmd/tf/model/ener.py", line 142, in data_stat
self._compute_input_stat(
File "/opt/deepmd-kit-3.0.0b4/lib/python3.10/site-packages/deepmd/tf/model/ener.py", line 150, in _compute_input_stat
self.descrpt.compute_input_stats(
TypeError: DescrptSeA.compute_input_stats() takes 7 positional arguments but 9 were given
Traceback (most recent call last):
File "/home/input_lbg-12166-15319264/script", line 40, in
output = op_obj.execute(input)
File "/home/input_lbg-12166-15319264/tmp/inputs/artifacts/dflow_python_packages/opt/deepmd-kit-3.0.0b3/lib/python3.10/site-packages/dflow/python/op.py", line 136, in wrapper_exec
op_out = func(self, op_in)
File "/home/input_lbg-12166-15319264/tmp/inputs/artifacts/dflow_python_packages/opt/deepmd-kit-3.0.0b3/lib/python3.10/site-packages/d
pgen2/op/run_dp_train.py", line 303, in execute
raise FatalError("dp train failed")
dflow.python.python_op_template.FatalError: dp train failed

[njzjz]

It seems that you feed the data with mixed types to a descriptor without mixed types, which is unsupported. I agree that the error message should be improved.