From abb3ca48c6e8913b2345e011b86a5370b72db1f3 Mon Sep 17 00:00:00 2001 From: Anyang Peng <137014849+anyangml@users.noreply.github.com> Date: Wed, 6 Mar 2024 20:21:15 +0800 Subject: [PATCH] fix: typo --- deepmd/pt/model/task/polarizability.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/deepmd/pt/model/task/polarizability.py b/deepmd/pt/model/task/polarizability.py index 783aed2614..98c9060704 100644 --- a/deepmd/pt/model/task/polarizability.py +++ b/deepmd/pt/model/task/polarizability.py @@ -114,7 +114,7 @@ def __init__( self.scale, dtype=env.GLOBAL_PT_FLOAT_PRECISION, device=env.DEVICE ).view(ntypes, 1) self.shift_diag = shift_diag - self.constant_matrix = torch.zero(self.ntypes) + self.constant_matrix = torch.zeros(self.ntypes) super().__init__( var_name=kwargs.pop("var_name", "polar"), ntypes=ntypes,