Skip to content

Commit f88ab82

Browse files
Fix UT
1 parent a1e2642 commit f88ab82

2 files changed

Lines changed: 2 additions & 2 deletions

File tree

deepmd/dpmodel/atomic_model/property_atomic_model.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -51,7 +51,7 @@ def apply_out_stat(
5151
5252
"""
5353
out_bias, out_std = self._fetch_out_stat(self.bias_keys)
54-
if self.get_compute_stats_distinguish_types:
54+
if self.get_compute_stats_distinguish_types():
5555
for kk in self.bias_keys:
5656
ret[kk] = ret[kk] * out_std[kk][0] + out_bias[kk][atype]
5757
else:

deepmd/pt/model/atomic_model/property_atomic_model.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -49,7 +49,7 @@ def apply_out_stat(
4949
5050
"""
5151
out_bias, out_std = self._fetch_out_stat(self.bias_keys)
52-
if self.get_compute_stats_distinguish_types:
52+
if self.get_compute_stats_distinguish_types():
5353
for kk in self.bias_keys:
5454
ret[kk] = ret[kk] * out_std[kk][0] + out_bias[kk][atype]
5555
else:

0 commit comments

Comments
 (0)