Fix: Pt Polar Stat keyword

[anyangml]

To fix #3582, should merge after #3571.

[codecov]

Codecov Report

All modified and coverable lines are covered by tests ✅

Project coverage is 77.71%. Comparing base (3c09d46) to head (ff7d3dd).

Additional details and impacted files

@@           Coverage Diff           @@
##            devel    #3584   +/-   ##
=======================================
  Coverage   77.70%   77.71%           
=======================================
  Files         433      433           
  Lines       37491    37498    +7     
  Branches     1620     1620           
=======================================
+ Hits        29134    29141    +7     
  Misses       7496     7496           
  Partials      861      861           

☔ View full report in Codecov by Sentry.
📢 Have feedback on the report? Share it here.

[anyangml]

There seems to be another bug KeyError: 'find_atomic_polarizability'.

I think the keys are removed here. @iProzd because the dtype for all the find_ terms are np.float32

deepmd-kit/deepmd/pt/utils/stat.py

Lines 56 to 62 in fb61efb

	sys_stat[dd] = None
	elif isinstance(stat_data[dd], torch.Tensor):
	if dd not in sys_stat:
	sys_stat[dd] = []
	sys_stat[dd].append(stat_data[dd])
	else:
	pass

deepmd-kit/deepmd/utils/data.py

Lines 666 to 678 in fb61efb

	if repeat != 1:
	data = np.repeat(data, repeat).reshape([nframes, -1])
	return np.float32(1.0), data
	elif must:
	raise RuntimeError("%s not found!" % path)
	else:
	if type_sel is not None and not output_natoms_for_type_sel:
	ndof = ndof_ * natoms_sel
	data = np.full([nframes, ndof], default, dtype=dtype)
	if repeat != 1:
	data = np.repeat(data, repeat).reshape([nframes, -1])
	return np.float32(0.0), data