Skip to content

Commit 24b301a

Browse files
committed
Revert "remove unused variable zbl_weight in DPZBLLinearEnergyAtomicModel"
This reverts commit 4932fbe.
1 parent 4932fbe commit 24b301a

1 file changed

Lines changed: 1 addition & 2 deletions

File tree

deepmd/dpmodel/atomic_model/linear_atomic_model.py

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -509,6 +509,5 @@ def _compute_weight(
509509
coef = xp.where(right_mask, xp.zeros_like(coef), coef)
510510
# to handle masked atoms
511511
coef = xp.where(sigma != 0, coef, xp.zeros_like(coef))
512-
# never used variable
513-
# self.zbl_weight = coef
512+
self.zbl_weight = coef
514513
return [1 - xp.expand_dims(coef, -1), xp.expand_dims(coef, -1)]

0 commit comments

Comments
 (0)